MP-1135 Upgrade pytorch

MoonVision · Apr 12, 2024 · 19c4621 · 19c4621
1 parent 60b9a00
commit 19c4621
Showing 1 changed file with 25 additions and 25 deletions.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -111,18 +111,18 @@ jobs:
     runs-on: ubuntu-latest
     steps:
     - uses: actions/checkout@v3
-#    - uses: ./.github/create-docker
-#      with:
-#        imagename: moonvision/${{ env.IMAGE_PRE }}custom-builds
-#        additionaltag: pytorch-3.11
-#        tagversion: 1.13.1_torchvision-0.15.2
-#        baseimage: moonvision/${{ env.IMAGE_PRE }}python-base:3.11-latest
-#        path: docker/builders/pytorch
-#        dockerfile: docker/builders/pytorch/Dockerfile
-#        additional_buildargs: |-
-#          pytorch_tag=v1.13.1
-#          torchvision_tag=v0.15.2
-#          prebuilt=false
+    - uses: ./.github/create-docker
+      with:
+        imagename: moonvision/${{ env.IMAGE_PRE }}custom-builds
+        additionaltag: pytorch-3.11
+        tagversion: 2.0.1_torchvision-0.15.2
+        baseimage: moonvision/${{ env.IMAGE_PRE }}python-base:3.11-latest
+        path: docker/builders/pytorch
+        dockerfile: docker/builders/pytorch/Dockerfile
+        additional_buildargs: |-
+          pytorch_tag=v2.0.1
+          torchvision_tag=v0.15.2
+          prebuilt=false
   build-pytorch-cuda:
     needs:
     - docker-python-base
@@ -148,19 +148,19 @@ jobs:
     runs-on: ubuntu-latest
     steps:
     - uses: actions/checkout@v3
-#    - uses: ./.github/create-docker
-#      with:
-#        imagename: moonvision/${{ env.IMAGE_PRE }}custom-builds
-#        additionaltag: pytorch-cuda-3.11
-#        tagversion: 1.13.1_torchvision-0.15.2
-#        baseimage: moonvision/${{ env.IMAGE_PRE }}python-base:3.11-latest
-#        path: docker/builders/pytorch
-#        dockerfile: docker/builders/pytorch/Dockerfile
-#        additional_buildargs: |-
-#          pytorch_tag=v1.13.1
-#          torchvision_tag=v0.15.2
-#          prebuilt=true
-#          with_cuda=true
+    - uses: ./.github/create-docker
+      with:
+        imagename: moonvision/${{ env.IMAGE_PRE }}custom-builds
+        additionaltag: pytorch-cuda-3.11
+        tagversion: 2.0.1_torchvision-0.15.2
+        baseimage: moonvision/${{ env.IMAGE_PRE }}python-base:3.11-latest
+        path: docker/builders/pytorch
+        dockerfile: docker/builders/pytorch/Dockerfile
+        additional_buildargs: |-
+          pytorch_tag=v2.0.1
+          torchvision_tag=v0.15.2
+          prebuilt=true
+          with_cuda=true
   docker-moonbox-mini:
     needs:
     - docker-python-base