FROM nvidia/cuda:11.1.1-devel-ubuntu18.04

WORKDIR /root