diff --git a/docker/cuda-10-1.dockerfile b/docker/cuda-10-1.dockerfile new file mode 100644 index 0000000..19909fa --- /dev/null +++ b/docker/cuda-10-1.dockerfile @@ -0,0 +1,45 @@ +ARG BASE=python:3.8 +FROM ${BASE} + +ENV CUDA_VERSION 10.1.243 +ENV CUDA_PKG_VERSION 10-1=$CUDA_VERSION-1 +ENV PATH /usr/local/nvidia/bin:/usr/local/cuda/bin:${PATH} +ENV LD_LIBRARY_PATH /usr/local/nvidia/lib:/usr/local/nvidia/lib64 + +ENV NVIDIA_VISIBLE_DEVICES all +ENV NVIDIA_DRIVER_CAPABILITIES compute,utility +ENV NVIDIA_REQUIRE_CUDA "cuda>=10.1 brand=tesla,driver>=396,driver<397 brand=tesla,driver>=410,driver<411 brand=tesla,driver>=418,driver<419" +ENV CUDNN_VERSION 7.6.5.32 +ENV NCCL_VERSION 2.7.8 + +LABEL com.nvidia.cuda.version="${CUDA_VERSION}" +LABEL com.nvidia.cudnn.version="${CUDNN_VERSION}" +LABEL com.nvidia.volumes.needed="nvidia_driver" + +RUN apt-get update \ + && apt-get install -y --no-install-recommends \ + gnupg2 \ + curl \ + ca-certificates \ + && curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub | apt-key add - \ + && echo "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64 /" > /etc/apt/sources.list.d/cuda.list \ + && echo "deb https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64 /" > /etc/apt/sources.list.d/nvidia-ml.list \ + && apt-get purge --autoremove -y curl \ + && apt-get update \ + && apt-get install -y --no-install-recommends \ + cuda-cudart-$CUDA_PKG_VERSION \ + cuda-compat-10-1 \ + && ln -s cuda-10.1 /usr/local/cuda \ + && echo "/usr/local/nvidia/lib" >> /etc/ld.so.conf.d/nvidia.conf \ + && echo "/usr/local/nvidia/lib64" >> /etc/ld.so.conf.d/nvidia.conf \ + && apt-get install -y --no-install-recommends \ + cuda-libraries-$CUDA_PKG_VERSION \ + cuda-npp-$CUDA_PKG_VERSION \ + cuda-nvtx-$CUDA_PKG_VERSION \ + libcublas10=10.2.1.243-1 \ + libcudnn7=$CUDNN_VERSION-1+cuda10.1 \ + libnccl2=$NCCL_VERSION-1+cuda10.1 \ + && apt-mark hold libnccl2 \ + && apt-mark hold libcudnn7 \ + && apt-mark hold libcublas10 \ + && rm -rf /var/lib/apt/lists/*