upgrade docker base image with cuDNN 9

huggingface · Nov 22, 2024 · 96987bf · 96987bf
1 parent c3afcb6
commit 96987bf
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/exporters/Dockerfile_exporters_gpu b/tests/exporters/Dockerfile_exporters_gpu
@@ -1,6 +1,6 @@
 # use version with cudnn 8.5 to match torch==1.13.1 that uses 8.5.0.96
 # has Python 3.8.10
-FROM nvcr.io/nvidia/tensorrt:23.05-py3
+FROM nvcr.io/nvidia/tensorrt:24.01-py3
 CMD nvidia-smi
 
 # Ignore interactive questions during `docker build`

diff --git a/tests/onnxruntime/docker/Dockerfile_onnxruntime_gpu b/tests/onnxruntime/docker/Dockerfile_onnxruntime_gpu
@@ -1,5 +1,5 @@
 # use version with CUDA 11.8 and TensorRT 8.5.1.7 to match ORT 1.14 requirements
-FROM nvcr.io/nvidia/tensorrt:23.05-py3
+FROM nvcr.io/nvidia/tensorrt:24.01-py3
 CMD nvidia-smi
 
 # Ignore interactive questions during `docker build`