@@ -25,85 +25,6 @@ RUN echo "tsflags=nodocs" | tee -a /etc/yum.conf && \
2525# End of OS Packages #
2626# ############################################
2727
28- # ##################################################################################################
29- # CUDA 12.1 Layer, from https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.1.1 #
30- # ##################################################################################################
31-
32- USER 0
33-
34- ENV NVARCH x86_64
35- ENV NVIDIA_REQUIRE_CUDA "cuda>=12.1 brand=tesla,driver>=470,driver<471 brand=unknown,driver>=470,driver<471 brand=nvidia,driver>=470,driver<471 brand=nvidiartx,driver>=470,driver<471 brand=geforce,driver>=470,driver<471 brand=geforcertx,driver>=470,driver<471 brand=quadro,driver>=470,driver<471 brand=quadrortx,driver>=470,driver<471 brand=titan,driver>=470,driver<471 brand=titanrtx,driver>=470,driver<471 brand=tesla,driver>=525,driver<526 brand=unknown,driver>=525,driver<526 brand=nvidia,driver>=525,driver<526 brand=nvidiartx,driver>=525,driver<526 brand=geforce,driver>=525,driver<526 brand=geforcertx,driver>=525,driver<526 brand=quadro,driver>=525,driver<526 brand=quadrortx,driver>=525,driver<526 brand=titan,driver>=525,driver<526 brand=titanrtx,driver>=525,driver<526"
36- ENV NV_CUDA_CUDART_VERSION 12.1.105-1
37-
38- COPY cuda.repo-x86_64 /etc/yum.repos.d/cuda.repo
39-
40- RUN NVIDIA_GPGKEY_SUM=d0664fbbdb8c32356d45de36c5984617217b2d0bef41b93ccecd326ba3b80c87 && \
41- curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/rhel9/${NVARCH}/D42D0685.pub | sed '/^Version/d' > /etc/pki/rpm-gpg/RPM-GPG-KEY-NVIDIA && \
42- echo "$NVIDIA_GPGKEY_SUM /etc/pki/rpm-gpg/RPM-GPG-KEY-NVIDIA" | sha256sum -c --strict -
43-
44- ENV CUDA_VERSION 12.1.1
45-
46- # For libraries in the cuda-compat-* package: https://docs.nvidia.com/cuda/eula/index.html#attachment-a
47- RUN yum upgrade -y && yum install -y \
48- cuda-cudart-12-1-${NV_CUDA_CUDART_VERSION} \
49- cuda-compat-12-1 \
50- && ln -s cuda-12.1 /usr/local/cuda \
51- && yum -y clean all --enablerepo='*' && \
52- rm -rf /var/cache/dnf && \
53- find /var/log -type f -name "*.log" -exec rm -f {} \;
54-
55- # nvidia-docker 1.0
56- RUN echo "/usr/local/nvidia/lib" >> /etc/ld.so.conf.d/nvidia.conf && \
57- echo "/usr/local/nvidia/lib64" >> /etc/ld.so.conf.d/nvidia.conf
58-
59- ENV PATH /usr/local/nvidia/bin:/usr/local/cuda/bin:${PATH}
60- ENV LD_LIBRARY_PATH /usr/local/nvidia/lib:/usr/local/nvidia/lib64
61-
62- COPY NGC-DL-CONTAINER-LICENSE /
63-
64- # nvidia-container-runtime
65- ENV NVIDIA_VISIBLE_DEVICES all
66- ENV NVIDIA_DRIVER_CAPABILITIES compute,utility
67-
68- ENV NV_CUDA_LIB_VERSION 12.1.1-1
69-
70- ENV NV_NVTX_VERSION 12.1.105-1
71- ENV NV_LIBNPP_VERSION 12.1.0.40-1
72- ENV NV_LIBNPP_PACKAGE libnpp-12-1-${NV_LIBNPP_VERSION}
73- ENV NV_LIBCUBLAS_VERSION 12.1.3.1-1
74- ENV NV_LIBNCCL_PACKAGE_NAME libnccl
75- ENV NV_LIBNCCL_PACKAGE_VERSION 2.17.1-1
76- ENV NV_LIBNCCL_VERSION 2.17.1
77- ENV NCCL_VERSION 2.17.1
78- ENV NV_LIBNCCL_PACKAGE ${NV_LIBNCCL_PACKAGE_NAME}-${NV_LIBNCCL_PACKAGE_VERSION}+cuda12.1
79-
80- RUN yum install -y \
81- cuda-libraries-12-1-${NV_CUDA_LIB_VERSION} \
82- cuda-nvtx-12-1-${NV_NVTX_VERSION} \
83- ${NV_LIBNPP_PACKAGE} \
84- libcublas-12-1-${NV_LIBCUBLAS_VERSION} \
85- ${NV_LIBNCCL_PACKAGE} \
86- && yum -y clean all --enablerepo='*' && \
87- rm -rf /var/cache/dnf && \
88- find /var/log -type f -name "*.log" -exec rm -f {} \;
89-
90- # Set this flag so that libraries can find the location of CUDA
91- ENV XLA_FLAGS=--xla_gpu_cuda_data_dir=/usr/local/cuda
92-
93- # CuDNN
94- ENV NV_CUDNN_VERSION 8.9.0.131-1
95- ENV NV_CUDNN_PACKAGE libcudnn8-${NV_CUDNN_VERSION}.cuda12.1
96-
97- RUN yum install -y \
98- ${NV_CUDNN_PACKAGE} \
99- && yum -y clean all --enablerepo='*' && \
100- rm -rf /var/cache/dnf && \
101- find /var/log -type f -name "*.log" -exec rm -f {} \;
102-
103- # ############################################
104- # End of CUDA 12.1 Layer #
105- # ############################################
106-
10728# #####################################
10829# Deploy Python packages and Jupyter #
10930# #####################################
0 commit comments