MP-1135 Update torchvision

MoonVision · Apr 11, 2024 · de8c8ce · de8c8ce
1 parent 73ad5c2
commit de8c8ce
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -115,13 +115,13 @@ jobs:
       with:
         imagename: moonvision/${{ env.IMAGE_PRE }}custom-builds
         additionaltag: pytorch-3.11
-        tagversion: 1.13.1_torchvision-0.12.0
+        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.12.0
+          torchvision_tag=v0.15.2
           prebuilt=false
   build-pytorch-cuda:
     needs:
@@ -152,13 +152,13 @@ jobs:
       with:
         imagename: moonvision/${{ env.IMAGE_PRE }}custom-builds
         additionaltag: pytorch-cuda-3.11
-        tagversion: 1.13.1_torchvision-0.12.0
+        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.12.0
+          torchvision_tag=v0.15.2
           prebuilt=true
           with_cuda=true
   docker-moonbox-mini: