Update docker files (#2102)

* update dockerfiles

* upgrade docker base image with cuDNN 9

Author: echarlaix

examples/onnxruntime/training/docker/Dockerfile-ort-nightly-cu118

@@ -74,4 +74,4 @@ RUN $PYTHON_EXE -m torch_ort.configure
 
 WORKDIR .
 
-CMD ["/bin/bash"]
+CMD ["/bin/bash"]

examples/onnxruntime/training/docker/Dockerfile-ort1.17.1-cu118

@@ -76,4 +76,4 @@ RUN pip uninstall nvidia-nccl-cu12 -y
 
 WORKDIR .
 
-CMD ["/bin/bash"]
+CMD ["/bin/bash"]

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:24.02-py3
 CMD nvidia-smi
 
 # Ignore interactive questions during `docker build`

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:24.02-py3
 CMD nvidia-smi
 
 # Ignore interactive questions during `docker build`

tests/onnxruntime/docker/Dockerfile_onnxruntime_trainer

@@ -80,4 +80,4 @@ RUN pip install /workspace/optimum[tests]
 
 ENV TEST_LEVEL=1
 CMD RUN_SLOW=1 pytest -v -rs onnxruntime/training/nightly_test_trainer.py --durations=0
-CMD RUN_SLOW=1 pytest -v -rs onnxruntime/training/nightly_test_examples.py --durations=0
+CMD RUN_SLOW=1 pytest -v -rs onnxruntime/training/nightly_test_examples.py --durations=0