bump docker image to use PT 2.0.1 and Lightning 2.0.4

speediedan · Jun 22, 2023 · c67338d · c67338d
1 parent 76db176
commit c67338d
Show file tree

Hide file tree

Showing 6 changed files with 6 additions and 6 deletions.
diff --git a/.azure-pipelines/gpu-tests.yml b/.azure-pipelines/gpu-tests.yml
@@ -48,7 +48,7 @@ jobs:
           image: "speediedan/finetuning-scheduler:py3.10-pt1.13.1-pl2.0-azpl-init"
           scope: "1x"
         'PyTorch | latest':
-          image: "speediedan/finetuning-scheduler:py3.10-pt2.0.0-pl2.0-azpl-init"
+          image: "speediedan/finetuning-scheduler:py3.10-pt2.0.1-pl2.0-azpl-init"
           scope: ""
     # how long to run the job before automatically cancelling
     timeoutInMinutes: "100"

diff --git a/.github/workflows/release-docker.yml b/.github/workflows/release-docker.yml
@@ -31,7 +31,7 @@ jobs:
       matrix:
         # initially building only the latest supported configuration
         python_version: ["3.10"]
-        pytorch_version: ["2.0.0"]
+        pytorch_version: ["2.0.1"]
         cust_base: ["cu11.8-"]
         pl_version: ["2.0"]
     steps:

diff --git a/dockers/base-cuda/Dockerfile b/dockers/base-cuda/Dockerfile
@@ -17,7 +17,7 @@ ARG OS_VER=ubuntu20.04
 FROM nvidia/cuda:${CUDA_VERSION}-devel-${OS_VER}
 
 ARG PYTHON_VERSION=3.10
-ARG PYTORCH_VERSION=2.0.0
+ARG PYTORCH_VERSION=2.0.1
 ARG CUST_BUILD=0
 ARG MKL_THREADING_LAYER=GNU
 

diff --git a/dockers/fts-az-base/Dockerfile b/dockers/fts-az-base/Dockerfile
@@ -11,7 +11,7 @@
 # limitations under the License.
 
 ARG PYTHON_VERSION=3.10
-ARG PYTORCH_VERSION=2.0.0
+ARG PYTORCH_VERSION=2.0.1
 ARG LIGHTNING_VERSION=2.0
 ARG CUST_BASE
 

diff --git a/dockers/release-conda/Dockerfile b/dockers/release-conda/Dockerfile
@@ -18,7 +18,7 @@ FROM nvidia/cuda:${CUDA_VERSION}-devel-ubuntu20.04
 
 ARG CUDATOOLKIT_VERSION=11.8
 ARG PYTHON_VERSION=3.10
-ARG PYTORCH_VERSION=2.0.0
+ARG PYTORCH_VERSION=2.0.1
 ARG CONDA_VERSION=4.13.0
 
 SHELL ["/bin/bash", "-c"]

diff --git a/dockers/release/Dockerfile b/dockers/release/Dockerfile
@@ -11,7 +11,7 @@
 # limitations under the License.
 
 ARG PYTHON_VERSION=3.10
-ARG PYTORCH_VERSION=2.0.0
+ARG PYTORCH_VERSION=2.0.1
 ARG LIGHTNING_VERSION=2.0
 ARG CUST_BASE