change: base image to ubuntu20.04

minostauros · May 24, 2024 · 69c159f · 69c159f
1 parent d713547
commit 69c159f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pytorch-2.1.2/torchvision-0.16.2/cuda-11.8.0/ffmpeg-6.1/opencv-4.9.0/python-3.10/Dockerfile b/pytorch-2.1.2/torchvision-0.16.2/cuda-11.8.0/ffmpeg-6.1/opencv-4.9.0/python-3.10/Dockerfile
@@ -20,7 +20,7 @@ ARG TARGET_CUDA_VERSION=cu118
 ARG CUDA_VERSION=11.8
 # TensorRT for cuda11.8 supports cuda11.x
 ARG TENSORRT_VERSION="8.6.1.6-1+cuda11.8"
-ARG ONNXRUNTIME_VERSION=1.15.1
+ARG ONNXRUNTIME_VERSION=1.18.0
 
 # environmental variable for apex
 # https://en.wikipedia.org/wiki/CUDA#GPUs_supported

diff --git a/pytorch-2.2.1/torchvision-0.17.1/cuda-11.8.0/ffmpeg-6.1/opencv-4.9.0/python-3.10/Dockerfile b/pytorch-2.2.1/torchvision-0.17.1/cuda-11.8.0/ffmpeg-6.1/opencv-4.9.0/python-3.10/Dockerfile
@@ -20,7 +20,7 @@ ARG TARGET_CUDA_VERSION=cu118
 ARG CUDA_VERSION=11.8
 # TensorRT for cuda11.8 supports cuda11.x
 ARG TENSORRT_VERSION="8.6.1.6-1+cuda11.8"
-ARG ONNXRUNTIME_VERSION=1.15.1
+ARG ONNXRUNTIME_VERSION=1.18.0
 
 # environmental variable for apex
 # https://en.wikipedia.org/wiki/CUDA#GPUs_supported