MP-1135 Upgrade pytorch for moonboxes

MoonVision · Apr 12, 2024 · e3e3d6e · e3e3d6e
1 parent 19c4621
commit e3e3d6e
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -196,7 +196,7 @@ jobs:
           baseimage: moonvision/${{ env.IMAGE_PRE }}python-base:3.11-latest
           additional_buildargs: |-
             ffmpeg_from_docker=moonvision/${{ env.IMAGE_PRE }}custom-builds:ffmpeg-4.2.1
-            pytorch_from_docker=moonvision/${{ env.IMAGE_PRE }}custom-builds:pytorch-3.11-1.13.1_torchvision-0.15.2
+            pytorch_from_docker=moonvision/${{ env.IMAGE_PRE }}custom-builds:pytorch-3.11-2.0.1_torchvision-0.15.2
   docker-moonbox-genicam:
     needs:
     - docker-python-base
@@ -234,7 +234,7 @@ jobs:
           additional_buildargs: |-
             with_genicam=true
             ffmpeg_from_docker=moonvision/${{ env.IMAGE_PRE }}custom-builds:ffmpeg-4.2.1
-            pytorch_from_docker=moonvision/${{ env.IMAGE_PRE }}custom-builds:pytorch-3.11-1.13.1_torchvision-0.15.2
+            pytorch_from_docker=moonvision/${{ env.IMAGE_PRE }}custom-builds:pytorch-3.11-2.0.1_torchvision-0.15.2
   docker-moonbox-basler:
     needs:
     - docker-python-base
@@ -274,7 +274,7 @@ jobs:
             with_genicam=true
             with_pylon=true
             ffmpeg_from_docker=moonvision/${{ env.IMAGE_PRE }}custom-builds:ffmpeg-4.2.1
-            pytorch_from_docker=moonvision/${{ env.IMAGE_PRE }}custom-builds:pytorch-3.11-1.13.1_torchvision-0.15.2
+            pytorch_from_docker=moonvision/${{ env.IMAGE_PRE }}custom-builds:pytorch-3.11-2.0.1_torchvision-0.15.2
   docker-moonbox-cuda:
     needs:
     - docker-python-base
@@ -316,4 +316,4 @@ jobs:
           with_pylon=false
           with_cuda=true
           ffmpeg_from_docker=moonvision/${{ env.IMAGE_PRE }}custom-builds:ffmpeg-cuda-4.2.1
-          pytorch_from_docker=moonvision/${{ env.IMAGE_PRE }}custom-builds:pytorch-cuda-3.11-1.13.1_torchvision-0.15.2
+          pytorch_from_docker=moonvision/${{ env.IMAGE_PRE }}custom-builds:pytorch-cuda-3.11-2.0.1_torchvision-0.15.2