change: base image to ubuntu20.04

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

Hide file tree

Showing 2 changed files with 6 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
@@ -75,6 +75,8 @@ RUN echo "Asia/Seoul" > /etc/timezone && \
     pip${PYTHON_VERSION} install h5py munkres scikit-learn visdom youtube_dl matplotlib nibabel pypng moviepy Pillow einops torchinfo && \
     pip${PYTHON_VERSION} install timm decord simplejson av psutil fire fairscale ftfy && \
 # nvidia apex
+# setuptools==69.5.1 for https://github.com/aws-neuron/aws-neuron-sdk/issues/893
+    pip${PYTHON_VERSION} install setuptools==69.5.1 && \
     DIR=/tmp/apex && \
     mkdir -p ${DIR} && \
     cd ${DIR} && \

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
@@ -75,6 +75,8 @@ RUN echo "Asia/Seoul" > /etc/timezone && \
     pip${PYTHON_VERSION} install h5py munkres scikit-learn visdom youtube_dl matplotlib nibabel pypng moviepy Pillow einops torchinfo && \
     pip${PYTHON_VERSION} install timm decord simplejson av psutil fire fairscale ftfy && \
 # nvidia apex
+# setuptools==69.5.1 for https://github.com/aws-neuron/aws-neuron-sdk/issues/893
+    pip${PYTHON_VERSION} install setuptools==69.5.1 && \
     DIR=/tmp/apex && \
     mkdir -p ${DIR} && \
     cd ${DIR} && \