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 index eb3f4d5..baa081d 100644 --- 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 index 433bb22..69965f8 100644 --- 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} && \