Skip to content

Commit da31add

Browse files
harshad16harshad16
authored
Merge pull request #454 from atheo89/RHOAIENG-4344
Update CUDA layer to 12.1
2 parents 2b94e30 + dc1d4d4 commit da31add

File tree

7 files changed

+164
-126
lines changed

7 files changed

+164
-126
lines changed

‎cuda/c9s-python-3.9/Dockerfile

Lines changed: 50 additions & 43 deletions
Original file line numberDiff line numberDiff line change
@@ -12,26 +12,26 @@ LABEL name="odh-notebook-cuda-c9s-python-3.9" \
1212
io.openshift.build.image="quay.io/opendatahub/workbench-images:cuda-c9s-python-3.9"
1313

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

1919
ENV NVARCH x86_64
20-
ENV NVIDIA_REQUIRE_CUDA "cuda>=11.8 brand=tesla,driver>=450,driver<451 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"
21-
ENV NV_CUDA_CUDART_VERSION 11.8.89-1
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
2222

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

2525
RUN NVIDIA_GPGKEY_SUM=d0664fbbdb8c32356d45de36c5984617217b2d0bef41b93ccecd326ba3b80c87 && \
26-
curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/rhel8/${NVARCH}/D42D0685.pub | sed '/^Version/d' > /etc/pki/rpm-gpg/RPM-GPG-KEY-NVIDIA && \
26+
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 11.8.0
29+
ENV CUDA_VERSION 12.1.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-11-8-${NV_CUDA_CUDART_VERSION} \
34-
cuda-compat-11-8 \
33+
cuda-cudart-12-1-${NV_CUDA_CUDART_VERSION} \
34+
cuda-compat-12-1 \
3535
&& yum clean all \
3636
&& rm -rf /var/cache/yum/*
3737

@@ -49,63 +49,70 @@ 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/11.8.0/ubi8/runtime/Dockerfile
53-
ENV NV_CUDA_LIB_VERSION 11.8.0-1
54-
ENV NV_NVTX_VERSION 11.8.86-1
55-
ENV NV_LIBNPP_VERSION 11.8.0.86-1
56-
ENV NV_LIBNPP_PACKAGE libnpp-11-8-${NV_LIBNPP_VERSION}
57-
ENV NV_LIBCUBLAS_VERSION 11.11.3.6-1
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
5858
ENV NV_LIBNCCL_PACKAGE_NAME libnccl
59-
ENV NV_LIBNCCL_PACKAGE_VERSION 2.15.5-1
60-
ENV NV_LIBNCCL_VERSION 2.15.5
61-
ENV NCCL_VERSION 2.15.5
62-
ENV NV_LIBNCCL_PACKAGE ${NV_LIBNCCL_PACKAGE_NAME}-${NV_LIBNCCL_PACKAGE_VERSION}+cuda11.8
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
6363

6464
RUN yum install -y \
65-
cuda-libraries-11-8-${NV_CUDA_LIB_VERSION} \
66-
cuda-nvtx-11-8-${NV_NVTX_VERSION} \
65+
cuda-libraries-12-1-${NV_CUDA_LIB_VERSION} \
66+
cuda-nvtx-12-1-${NV_NVTX_VERSION} \
6767
${NV_LIBNPP_PACKAGE} \
68-
libcublas-11-8-${NV_LIBCUBLAS_VERSION} \
68+
libcublas-12-1-${NV_LIBCUBLAS_VERSION} \
6969
${NV_LIBNCCL_PACKAGE} \
7070
&& yum clean all \
7171
&& rm -rf /var/cache/yum/*
7272

73+
# Set this flag so that libraries can find the location of CUDA
74+
ENV XLA_FLAGS=--xla_gpu_cuda_data_dir=/usr/local/cuda
75+
7376
# Install CUDA devel from:
74-
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/11.8.0/ubi8/devel/Dockerfile
75-
ENV NV_CUDA_LIB_VERSION 11.8.0-1
76-
ENV NV_NVPROF_VERSION 11.8.87-1
77-
ENV NV_NVPROF_DEV_PACKAGE cuda-nvprof-11-8-${NV_NVPROF_VERSION}
78-
ENV NV_CUDA_CUDART_DEV_VERSION 11.8.89-1
79-
ENV NV_NVML_DEV_VERSION 11.8.86-1
80-
ENV NV_LIBCUBLAS_DEV_VERSION 11.11.3.6-1
81-
ENV NV_LIBNPP_DEV_VERSION 11.8.0.86-1
82-
ENV NV_LIBNPP_DEV_PACKAGE libnpp-devel-11-8-${NV_LIBNPP_DEV_VERSION}
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}
8386
ENV NV_LIBNCCL_DEV_PACKAGE_NAME libnccl-devel
84-
ENV NV_LIBNCCL_DEV_PACKAGE_VERSION 2.15.5-1
85-
ENV NCCL_VERSION 2.15.5
86-
ENV NV_LIBNCCL_DEV_PACKAGE ${NV_LIBNCCL_DEV_PACKAGE_NAME}-${NV_LIBNCCL_DEV_PACKAGE_VERSION}+cuda11.8
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}
8792

8893
RUN yum install -y \
8994
make \
90-
cuda-command-line-tools-11-8-${NV_CUDA_LIB_VERSION} \
91-
cuda-libraries-devel-11-8-${NV_CUDA_LIB_VERSION} \
92-
cuda-minimal-build-11-8-${NV_CUDA_LIB_VERSION} \
93-
cuda-cudart-devel-11-8-${NV_CUDA_CUDART_DEV_VERSION} \
95+
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} \
94100
${NV_NVPROF_DEV_PACKAGE} \
95-
cuda-nvml-devel-11-8-${NV_NVML_DEV_VERSION} \
96-
libcublas-devel-11-8-${NV_LIBCUBLAS_DEV_VERSION} \
101+
cuda-nvml-devel-12-1-${NV_NVML_DEV_VERSION} \
102+
libcublas-devel-12-1-${NV_LIBCUBLAS_DEV_VERSION} \
97103
${NV_LIBNPP_DEV_PACKAGE} \
98104
${NV_LIBNCCL_DEV_PACKAGE} \
105+
${NV_CUDA_NSIGHT_COMPUTE_DEV_PACKAGE} \
99106
&& yum clean all \
100107
&& rm -rf /var/cache/yum/*
101108

102109
ENV LIBRARY_PATH /usr/local/cuda/lib64/stubs
103110

104111
# Install CUDA devel cudnn8 from:
105-
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/11.8.0/ubi8/devel/cudnn8/Dockerfile
112+
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.1.1/ubi9/devel/cudnn8/Dockerfile
106113
ENV NV_CUDNN_VERSION 8.9.0.131-1
107-
ENV NV_CUDNN_PACKAGE libcudnn8-${NV_CUDNN_VERSION}.cuda11.8
108-
ENV NV_CUDNN_PACKAGE_DEV libcudnn8-devel-${NV_CUDNN_VERSION}.cuda11.8
114+
ENV NV_CUDNN_PACKAGE libcudnn8-${NV_CUDNN_VERSION}.cuda12.1
115+
ENV NV_CUDNN_PACKAGE_DEV libcudnn8-devel-${NV_CUDNN_VERSION}.cuda12.1
109116

110117
LABEL com.nvidia.cudnn.version="${NV_CUDNN_VERSION}"
111118

@@ -115,8 +122,8 @@ RUN yum install -y \
115122
&& yum clean all \
116123
&& rm -rf /var/cache/yum/*
117124

118-
# Install CUDA toolkit 11.8
119-
RUN yum -y install cuda-toolkit-11-8 && \
125+
# Install CUDA toolkit 12.1
126+
RUN yum -y install cuda-toolkit-12-1 && \
120127
yum -y clean all --enablerepo="*"
121128

122129
# Set this flag so that libraries can find the location of CUDA

‎cuda/c9s-python-3.9/cuda.repo-arm64

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
[cuda]
2+
name=cuda
3+
baseurl=https://developer.download.nvidia.com/compute/cuda/repos/rhel9/sbsa
4+
enabled=1
5+
gpgcheck=1
6+
gpgkey=file:///etc/pki/rpm-gpg/RPM-GPG-KEY-NVIDIA

‎cuda/ubi8-python-3.8/Dockerfile

Lines changed: 45 additions & 42 deletions
Original file line numberDiff line numberDiff line change
@@ -12,27 +12,26 @@ LABEL name="odh-notebook-cuda-ubi8-python-3.8" \
1212
io.openshift.build.image="quay.io/opendatahub/workbench-images:cuda-ubi8-python-3.8"
1313

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

1919
ENV NVARCH x86_64
20-
ENV NVIDIA_REQUIRE_CUDA "cuda>=11.4 brand=tesla,driver>=418,driver<419 brand=tesla,driver>=450,driver<451"
21-
ENV NV_CUDA_CUDART_VERSION 11.4.108-1
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
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/rhel8/${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 11.4.2
29+
ENV CUDA_VERSION 12.1.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-11-4-${NV_CUDA_CUDART_VERSION} \
34-
cuda-compat-11-4 \
35-
&& ln -s cuda-11.4 /usr/local/cuda \
33+
cuda-cudart-12-1-${NV_CUDA_CUDART_VERSION} \
34+
cuda-compat-12-1 \
3635
&& yum clean all \
3736
&& rm -rf /var/cache/yum/*
3837

@@ -50,63 +49,67 @@ ENV NVIDIA_VISIBLE_DEVICES all
5049
ENV NVIDIA_DRIVER_CAPABILITIES compute,utility
5150

5251
# Install CUDA runtime from:
53-
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/11.4.2/ubi8/runtime/Dockerfile
54-
ENV NV_CUDA_LIB_VERSION 11.4.2-1
55-
ENV NV_NVTX_VERSION 11.4.120-1
56-
ENV NV_LIBNPP_VERSION 11.4.0.110-1
57-
ENV NV_LIBNPP_PACKAGE libnpp-11-4-${NV_LIBNPP_VERSION}
58-
ENV NV_LIBCUBLAS_VERSION 11.6.1.51-1
52+
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.1.1/ubi8/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
5958
ENV NV_LIBNCCL_PACKAGE_NAME libnccl
60-
ENV NV_LIBNCCL_PACKAGE_VERSION 2.11.4-1
61-
ENV NV_LIBNCCL_VERSION 2.11.4
62-
ENV NCCL_VERSION 2.11.4
63-
ENV NV_LIBNCCL_PACKAGE ${NV_LIBNCCL_PACKAGE_NAME}-${NV_LIBNCCL_PACKAGE_VERSION}+cuda11.4
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
6463

6564
RUN yum install -y \
66-
cuda-libraries-11-4-${NV_CUDA_LIB_VERSION} \
67-
cuda-nvtx-11-4-${NV_NVTX_VERSION} \
65+
cuda-libraries-12-1-${NV_CUDA_LIB_VERSION} \
66+
cuda-nvtx-12-1-${NV_NVTX_VERSION} \
6867
${NV_LIBNPP_PACKAGE} \
69-
libcublas-11-4-${NV_LIBCUBLAS_VERSION} \
68+
libcublas-12-1-${NV_LIBCUBLAS_VERSION} \
7069
${NV_LIBNCCL_PACKAGE} \
7170
&& yum clean all \
7271
&& rm -rf /var/cache/yum/*
7372

7473
# Install CUDA devel from:
75-
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/11.4.2/ubi8/devel/Dockerfile
76-
ENV NV_CUDA_LIB_VERSION 11.4.2-1
77-
ENV NV_NVPROF_VERSION 11.4.120-1
78-
ENV NV_NVPROF_DEV_PACKAGE cuda-nvprof-11-4-${NV_NVPROF_VERSION}
79-
ENV NV_CUDA_CUDART_DEV_VERSION 11.4.108-1
80-
ENV NV_NVML_DEV_VERSION 11.4.120-1
81-
ENV NV_LIBCUBLAS_DEV_VERSION 11.6.1.51-1
82-
ENV NV_LIBNPP_DEV_VERSION 11.4.0.110-1
83-
ENV NV_LIBNPP_DEV_PACKAGE libnpp-devel-11-4-${NV_LIBNPP_DEV_VERSION}
74+
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.1.1/ubi8/devel/Dockerfile
75+
ENV NV_CUDA_LIB_VERSION 12.1.1-1
76+
ENV NV_NVPROF_VERSION 12.1.105-1
77+
ENV NV_NVPROF_DEV_PACKAGE cuda-nvprof-12-1-${NV_NVPROF_VERSION}
78+
ENV NV_CUDA_CUDART_DEV_VERSION 12.1.105-1
79+
ENV NV_NVML_DEV_VERSION 12.1.105-1
80+
ENV NV_LIBCUBLAS_DEV_VERSION 12.1.3.1-1
81+
ENV NV_LIBNPP_DEV_VERSION 12.1.0.40-1
82+
ENV NV_LIBNPP_DEV_PACKAGE libnpp-devel-12-1-${NV_LIBNPP_DEV_VERSION}
8483
ENV NV_LIBNCCL_DEV_PACKAGE_NAME libnccl-devel
85-
ENV NV_LIBNCCL_DEV_PACKAGE_VERSION 2.11.4-1
86-
ENV NCCL_VERSION 2.11.4
87-
ENV NV_LIBNCCL_DEV_PACKAGE ${NV_LIBNCCL_DEV_PACKAGE_NAME}-${NV_LIBNCCL_DEV_PACKAGE_VERSION}+cuda11.4
84+
ENV NV_LIBNCCL_DEV_PACKAGE_VERSION 2.17.1-1
85+
ENV NCCL_VERSION 2.17.1
86+
ENV NV_LIBNCCL_DEV_PACKAGE ${NV_LIBNCCL_DEV_PACKAGE_NAME}-${NV_LIBNCCL_DEV_PACKAGE_VERSION}+cuda12.1
87+
ENV NV_CUDA_NSIGHT_COMPUTE_VERSION 12.1.1-1
88+
ENV NV_CUDA_NSIGHT_COMPUTE_DEV_PACKAGE cuda-nsight-compute-12-1-${NV_CUDA_NSIGHT_COMPUTE_VERSION}
8889

8990
RUN yum install -y \
9091
make \
91-
cuda-command-line-tools-11-4-${NV_CUDA_LIB_VERSION} \
92-
cuda-libraries-devel-11-4-${NV_CUDA_LIB_VERSION} \
93-
cuda-minimal-build-11-4-${NV_CUDA_LIB_VERSION} \
94-
cuda-cudart-devel-11-4-${NV_CUDA_CUDART_DEV_VERSION} \
92+
findutils \
93+
cuda-command-line-tools-12-1-${NV_CUDA_LIB_VERSION} \
94+
cuda-libraries-devel-12-1-${NV_CUDA_LIB_VERSION} \
95+
cuda-minimal-build-12-1-${NV_CUDA_LIB_VERSION} \
96+
cuda-cudart-devel-12-1-${NV_CUDA_CUDART_DEV_VERSION} \
9597
${NV_NVPROF_DEV_PACKAGE} \
96-
cuda-nvml-devel-11-4-${NV_NVML_DEV_VERSION} \
97-
libcublas-devel-11-4-${NV_LIBCUBLAS_DEV_VERSION} \
98+
cuda-nvml-devel-12-1-${NV_NVML_DEV_VERSION} \
99+
libcublas-devel-12-1-${NV_LIBCUBLAS_DEV_VERSION} \
98100
${NV_LIBNPP_DEV_PACKAGE} \
99101
${NV_LIBNCCL_DEV_PACKAGE} \
102+
${NV_CUDA_NSIGHT_COMPUTE_DEV_PACKAGE} \
100103
&& yum clean all \
101104
&& rm -rf /var/cache/yum/*
102105

103106
ENV LIBRARY_PATH /usr/local/cuda/lib64/stubs
104107

105108
# Install CUDA devel cudnn8 from:
106-
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/11.4.2/ubi8/devel/cudnn8/Dockerfile
107-
ENV NV_CUDNN_VERSION 8.2.4.15-1
108-
ENV NV_CUDNN_PACKAGE libcudnn8-${NV_CUDNN_VERSION}.cuda11.4
109-
ENV NV_CUDNN_PACKAGE_DEV libcudnn8-devel-${NV_CUDNN_VERSION}.cuda11.4
109+
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.1.1/ubi8/devel/cudnn8/Dockerfile
110+
ENV NV_CUDNN_VERSION 8.9.0.131-1
111+
ENV NV_CUDNN_PACKAGE libcudnn8-${NV_CUDNN_VERSION}.cuda12.1
112+
ENV NV_CUDNN_PACKAGE_DEV libcudnn8-devel-${NV_CUDNN_VERSION}.cuda12.1
110113

111114
LABEL com.nvidia.cudnn.version="${NV_CUDNN_VERSION}"
112115

‎cuda/ubi8-python-3.8/cuda.repo-arm64

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
[cuda]
2+
name=cuda
3+
baseurl=https://developer.download.nvidia.com/compute/cuda/repos/rhel8/sbsa
4+
enabled=1
5+
gpgcheck=1
6+
gpgkey=file:///etc/pki/rpm-gpg/RPM-GPG-KEY-NVIDIA

‎cuda/ubi8-python-3.8/cuda.repo-ppc64le

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
[cuda]
2+
name=cuda
3+
baseurl=https://developer.download.nvidia.com/compute/cuda/repos/rhel8/ppc64le
4+
enabled=1
5+
gpgcheck=1
6+
gpgkey=file:///etc/pki/rpm-gpg/RPM-GPG-KEY-NVIDIA

0 commit comments

Comments
 (0)