Skip to content
Snippets Groups Projects
Unverified Commit 9f813882 authored by Henning Becker's avatar Henning Becker Committed by GitHub
Browse files

Update CUDA to 12.3 in tf_runtime Docker images (#165)

parent 3521d586
No related branches found
No related tags found
No related merge requests found
...@@ -13,7 +13,7 @@ ...@@ -13,7 +13,7 @@
# limitations under the License. # limitations under the License.
# ============================================================================ # ============================================================================
FROM nvidia/cuda:11.8.0-base-ubuntu22.04 as base FROM nvidia/cuda:12.3.0-base-ubuntu22.04 as base
ENV DEBIAN_FRONTEND=noninteractive ENV DEBIAN_FRONTEND=noninteractive
ENV LANG C.UTF-8 ENV LANG C.UTF-8
...@@ -52,5 +52,5 @@ CMD ["bash", "-c", "source /etc/bash.bashrc && jupyter notebook --notebook-dir=/ ...@@ -52,5 +52,5 @@ CMD ["bash", "-c", "source /etc/bash.bashrc && jupyter notebook --notebook-dir=/
FROM base as test FROM base as test
ENV LD_LIBRARY_PATH /usr/local/cuda/lib64/stubs/:$LD_LIBRARY_PATH ENV LD_LIBRARY_PATH /usr/local/cuda/lib64/stubs/:$LD_LIBRARY_PATH
COPY test.import_cpu.sh /test.import_cpu.sh COPY test.import_gpu.sh /test.import_gpu.sh
RUN /test.import_cpu.sh RUN /test.import_gpu.sh
# All required CUDA packages # All required CUDA packages
cuda-command-line-tools-11-8 cuda-command-line-tools-12-3
cuda-cudart-dev-11-8 cuda-cudart-dev-12-3
cuda-nvcc-11-8 cuda-nvcc-12-3
cuda-cupti-11-8 cuda-cupti-12-3
cuda-nvprune-11-8 cuda-nvprune-12-3
cuda-libraries-11-8 cuda-libraries-12-3
cuda-nvrtc-11-8 cuda-nvrtc-12-3
libcufft-11-8 libcufft-12-3
libcurand-11-8 libcurand-12-3
libcusolver-11-8 libcusolver-12-3
libcusparse-11-8 libcusparse-12-3
libcublas-11-8 libcublas-12-3
# CuDNN: https://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html#ubuntu-network-installation # CuDNN: https://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html#ubuntu-network-installation
libcudnn8=8.6.0.163-1+cuda11.8 libcudnn8=8.9.6.50-1+cuda12.2
libnvinfer-plugin8=8.6.1.6-1+cuda11.8 libnvinfer-plugin8=8.6.1.6-1+cuda12.0
libnvinfer8=8.6.1.6-1+cuda11.8 libnvinfer8=8.6.1.6-1+cuda12.0
# Other packages # Other packages
build-essential build-essential
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment