Skip to content

Commit 43efedd

Browse files
atheo89atheo89
committed
Upgrade cuda drivers from 12.1 to 12.4
1 parent 22e395c commit 43efedd

File tree

1 file changed

+45
-45
lines changed

1 file changed

+45
-45
lines changed

‎cuda/ubi9-python-3.11/Dockerfile

Lines changed: 45 additions & 45 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ FROM ${BASE_IMAGE}
33

44
LABEL name="odh-notebook-cuda-ubi9-python-3.11" \
55
summary="CUDA Python 3.11 base image for ODH notebooks" \
6-
description="CUDA Python 3.11 builder image based on UBI8 for ODH notebooks" \
6+
description="CUDA Python 3.11 builder image based on UBI9 for ODH notebooks" \
77
io.k8s.display-name="CUDA Python 3.11 base image for ODH notebooks" \
88
io.k8s.description="CUDA Python 3.11 builder image based on UBI8 for ODH notebooks" \
99
authoritative-source-url="https://github.com/opendatahub-io/notebooks" \
@@ -12,26 +12,26 @@ LABEL name="odh-notebook-cuda-ubi9-python-3.11" \
1212
io.openshift.build.image="quay.io/opendatahub/workbench-images:cuda-ubi9-python-3.11"
1313

1414
# Install CUDA base from:
15-
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.1.1/ubi9/base/Dockerfile
15+
# https://gitlab.com/nvidia/container-images/cuda/-/tree/master/dist/12.4.1/ubi9/base
1616
USER 0
1717
WORKDIR /opt/app-root/bin
1818

19-
ENV NVARCH x86_64
20-
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"
21-
ENV NV_CUDA_CUDART_VERSION 12.1.105-1
19+
ENV NVARCH sbsa
20+
ENV NVIDIA_REQUIRE_CUDA "cuda>=12.4 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 brand=tesla,driver>=535,driver<536 brand=unknown,driver>=535,driver<536 brand=nvidia,driver>=535,driver<536 brand=nvidiartx,driver>=535,driver<536 brand=geforce,driver>=535,driver<536 brand=geforcertx,driver>=535,driver<536 brand=quadro,driver>=535,driver<536 brand=quadrortx,driver>=535,driver<536 brand=titan,driver>=535,driver<536 brand=titanrtx,driver>=535,driver<536"
21+
ENV NV_CUDA_CUDART_VERSION 12.4.127-1
2222

2323
COPY cuda.repo-x86_64 /etc/yum.repos.d/cuda.repo
2424

2525
RUN NVIDIA_GPGKEY_SUM=d0664fbbdb8c32356d45de36c5984617217b2d0bef41b93ccecd326ba3b80c87 && \
2626
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 && \
2727
echo "$NVIDIA_GPGKEY_SUM /etc/pki/rpm-gpg/RPM-GPG-KEY-NVIDIA" | sha256sum -c --strict -
2828

29-
ENV CUDA_VERSION 12.1.1
29+
ENV CUDA_VERSION 12.4.1
3030

3131
# For libraries in the cuda-compat-* package: https://docs.nvidia.com/cuda/eula/index.html#attachment-a
3232
RUN yum upgrade -y && yum install -y \
33-
cuda-cudart-12-1-${NV_CUDA_CUDART_VERSION} \
34-
cuda-compat-12-1 \
33+
cuda-cudart-12-4-${NV_CUDA_CUDART_VERSION} \
34+
cuda-compat-12-4 \
3535
&& yum clean all \
3636
&& rm -rf /var/cache/yum/*
3737

@@ -49,23 +49,23 @@ ENV NVIDIA_VISIBLE_DEVICES all
4949
ENV NVIDIA_DRIVER_CAPABILITIES compute,utility
5050

5151
# Install CUDA runtime from:
52-
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.1.1/ubi9/runtime/Dockerfile
53-
ENV NV_CUDA_LIB_VERSION 12.1.1-1
54-
ENV NV_NVTX_VERSION 12.1.105-1
55-
ENV NV_LIBNPP_VERSION 12.1.0.40-1
56-
ENV NV_LIBNPP_PACKAGE libnpp-12-1-${NV_LIBNPP_VERSION}
57-
ENV NV_LIBCUBLAS_VERSION 12.1.3.1-1
52+
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.4.1/ubi9/runtime/Dockerfile
53+
ENV NV_CUDA_LIB_VERSION 12.4.1-1
54+
ENV NV_NVTX_VERSION 12.4.127-1
55+
ENV NV_LIBNPP_VERSION 12.2.5.30-1
56+
ENV NV_LIBNPP_PACKAGE libnpp-12-4-${NV_LIBNPP_VERSION}
57+
ENV NV_LIBCUBLAS_VERSION 12.4.5.8-1
5858
ENV NV_LIBNCCL_PACKAGE_NAME libnccl
59-
ENV NV_LIBNCCL_PACKAGE_VERSION 2.17.1-1
60-
ENV NV_LIBNCCL_VERSION 2.17.1
61-
ENV NCCL_VERSION 2.17.1
62-
ENV NV_LIBNCCL_PACKAGE ${NV_LIBNCCL_PACKAGE_NAME}-${NV_LIBNCCL_PACKAGE_VERSION}+cuda12.1
59+
ENV NV_LIBNCCL_PACKAGE_VERSION 2.21.5-1
60+
ENV NV_LIBNCCL_VERSION 2.21.5
61+
ENV NCCL_VERSION 2.21.5
62+
ENV NV_LIBNCCL_PACKAGE ${NV_LIBNCCL_PACKAGE_NAME}-${NV_LIBNCCL_PACKAGE_VERSION}+cuda12.4
6363

6464
RUN yum install -y \
65-
cuda-libraries-12-1-${NV_CUDA_LIB_VERSION} \
66-
cuda-nvtx-12-1-${NV_NVTX_VERSION} \
65+
cuda-libraries-12-4-${NV_CUDA_LIB_VERSION} \
66+
cuda-nvtx-12-4-${NV_NVTX_VERSION} \
6767
${NV_LIBNPP_PACKAGE} \
68-
libcublas-12-1-${NV_LIBCUBLAS_VERSION} \
68+
libcublas-12-4-${NV_LIBCUBLAS_VERSION} \
6969
${NV_LIBNCCL_PACKAGE} \
7070
&& yum clean all \
7171
&& rm -rf /var/cache/yum/*
@@ -74,32 +74,32 @@ RUN yum install -y \
7474
ENV XLA_FLAGS=--xla_gpu_cuda_data_dir=/usr/local/cuda
7575

7676
# Install CUDA devel from:
77-
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.1.1/ubi9/devel/Dockerfile
78-
ENV NV_CUDA_LIB_VERSION 12.1.1-1
79-
ENV NV_NVPROF_VERSION 12.1.105-1
80-
ENV NV_NVPROF_DEV_PACKAGE cuda-nvprof-12-1-${NV_NVPROF_VERSION}
81-
ENV NV_CUDA_CUDART_DEV_VERSION 12.1.105-1
82-
ENV NV_NVML_DEV_VERSION 12.1.105-1
83-
ENV NV_LIBCUBLAS_DEV_VERSION 12.1.3.1-1
84-
ENV NV_LIBNPP_DEV_VERSION 12.1.0.40-1
85-
ENV NV_LIBNPP_DEV_PACKAGE libnpp-devel-12-1-${NV_LIBNPP_DEV_VERSION}
77+
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.4.1/ubi9/devel/Dockerfile
78+
ENV NV_CUDA_LIB_VERSION 12.4.1-1
79+
ENV NV_NVPROF_VERSION 12.4.127-1
80+
ENV NV_NVPROF_DEV_PACKAGE cuda-nvprof-12-4-${NV_NVPROF_VERSION}
81+
ENV NV_CUDA_CUDART_DEV_VERSION 12.4.127-1
82+
ENV NV_NVML_DEV_VERSION 12.4.127-1
83+
ENV NV_LIBCUBLAS_DEV_VERSION 12.4.5.8-1
84+
ENV NV_LIBNPP_DEV_VERSION 12.2.5.30-1
85+
ENV NV_LIBNPP_DEV_PACKAGE libnpp-devel-12-4-${NV_LIBNPP_DEV_VERSION}
8686
ENV NV_LIBNCCL_DEV_PACKAGE_NAME libnccl-devel
87-
ENV NV_LIBNCCL_DEV_PACKAGE_VERSION 2.17.1-1
88-
ENV NCCL_VERSION 2.17.1
89-
ENV NV_LIBNCCL_DEV_PACKAGE ${NV_LIBNCCL_DEV_PACKAGE_NAME}-${NV_LIBNCCL_DEV_PACKAGE_VERSION}+cuda12.1
90-
ENV NV_CUDA_NSIGHT_COMPUTE_VERSION 12.1.1-1
91-
ENV NV_CUDA_NSIGHT_COMPUTE_DEV_PACKAGE cuda-nsight-compute-12-1-${NV_CUDA_NSIGHT_COMPUTE_VERSION}
87+
ENV NV_LIBNCCL_DEV_PACKAGE_VERSION 2.21.5-1
88+
ENV NCCL_VERSION 2.21.5
89+
ENV NV_LIBNCCL_DEV_PACKAGE ${NV_LIBNCCL_DEV_PACKAGE_NAME}-${NV_LIBNCCL_DEV_PACKAGE_VERSION}+cuda12.4
90+
ENV NV_CUDA_NSIGHT_COMPUTE_VERSION 12.4.1-1
91+
ENV NV_CUDA_NSIGHT_COMPUTE_DEV_PACKAGE cuda-nsight-compute-12-4-${NV_CUDA_NSIGHT_COMPUTE_VERSION}
9292

9393
RUN yum install -y \
9494
make \
9595
findutils \
96-
cuda-command-line-tools-12-1-${NV_CUDA_LIB_VERSION} \
97-
cuda-libraries-devel-12-1-${NV_CUDA_LIB_VERSION} \
98-
cuda-minimal-build-12-1-${NV_CUDA_LIB_VERSION} \
99-
cuda-cudart-devel-12-1-${NV_CUDA_CUDART_DEV_VERSION} \
96+
cuda-command-line-tools-12-4-${NV_CUDA_LIB_VERSION} \
97+
cuda-libraries-devel-12-4-${NV_CUDA_LIB_VERSION} \
98+
cuda-minimal-build-12-4-${NV_CUDA_LIB_VERSION} \
99+
cuda-cudart-devel-12-4-${NV_CUDA_CUDART_DEV_VERSION} \
100100
${NV_NVPROF_DEV_PACKAGE} \
101-
cuda-nvml-devel-12-1-${NV_NVML_DEV_VERSION} \
102-
libcublas-devel-12-1-${NV_LIBCUBLAS_DEV_VERSION} \
101+
cuda-nvml-devel-12-4-${NV_NVML_DEV_VERSION} \
102+
libcublas-devel-12-4-${NV_LIBCUBLAS_DEV_VERSION} \
103103
${NV_LIBNPP_DEV_PACKAGE} \
104104
${NV_LIBNCCL_DEV_PACKAGE} \
105105
${NV_CUDA_NSIGHT_COMPUTE_DEV_PACKAGE} \
@@ -109,10 +109,10 @@ RUN yum install -y \
109109
ENV LIBRARY_PATH /usr/local/cuda/lib64/stubs
110110

111111
# Install CUDA devel cudnn8 from:
112-
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.1.1/ubi9/devel/cudnn8/Dockerfile
113-
ENV NV_CUDNN_VERSION 8.9.0.131-1
114-
ENV NV_CUDNN_PACKAGE libcudnn8-${NV_CUDNN_VERSION}.cuda12.1
115-
ENV NV_CUDNN_PACKAGE_DEV libcudnn8-devel-${NV_CUDNN_VERSION}.cuda12.1
112+
# hhttps://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.4.1/ubi9/devel/cudnn/Dockerfile
113+
ENV NV_CUDNN_VERSION 9.1.0.70-1
114+
ENV NV_CUDNN_PACKAGE libcudnn9-cuda-12-${NV_CUDNN_VERSION}
115+
ENV NV_CUDNN_PACKAGE_DEV libcudnn9-devel-cuda-12-${NV_CUDNN_VERSION}
116116

117117
LABEL com.nvidia.cudnn.version="${NV_CUDNN_VERSION}"
118118

0 commit comments

Comments
 (0)