update dockerfiles

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

Hide file tree

Showing 5 changed files with 5 additions and 5 deletions.
diff --git a/examples/onnxruntime/training/docker/Dockerfile-ort-nightly-cu118 b/examples/onnxruntime/training/docker/Dockerfile-ort-nightly-cu118
@@ -15,7 +15,7 @@
 # limitations under the License.
 
 # Use nvidia/cuda image
-FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu20.04
+FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04
 CMD nvidia-smi
 
 # Ignore interactive questions during `docker build`

diff --git a/examples/onnxruntime/training/docker/Dockerfile-ort1.17.1-cu118 b/examples/onnxruntime/training/docker/Dockerfile-ort1.17.1-cu118
@@ -15,7 +15,7 @@
 # limitations under the License.
 
 # Use nvidia/cuda image
-FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu20.04
+FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04
 CMD nvidia-smi
 
 # Ignore interactive questions during `docker build`

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:22.08-py3
+FROM nvcr.io/nvidia/tensorrt:23.05-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:22.12-py3
+FROM nvcr.io/nvidia/tensorrt:23.05-py3
 CMD nvidia-smi
 
 # Ignore interactive questions during `docker build`

diff --git a/tests/onnxruntime/docker/Dockerfile_onnxruntime_trainer b/tests/onnxruntime/docker/Dockerfile_onnxruntime_trainer
@@ -15,7 +15,7 @@
 # limitations under the License.
 
 # Use nvidia/cuda image
-FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu20.04
+FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04
 CMD nvidia-smi
 
 # Ignore interactive questions during `docker build`