MP-1215 Use prebuilt pytorch for cpu images

MoonVision · Nov 26, 2024 · 2307864 · 2307864
1 parent d4fbda4
commit 2307864
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -123,7 +123,7 @@ jobs:
         additional_buildargs: |-
           pytorch_tag=v2.0.1
           torchvision_tag=v0.15.2
-          prebuilt=false
+          prebuilt=true
   build-pytorch-311:
     needs:
     - docker-python-base
@@ -141,7 +141,7 @@ jobs:
         additional_buildargs: |-
           pytorch_tag=v2.0.1
           torchvision_tag=v0.15.2
-          prebuilt=false
+          prebuilt=true
   build-pytorch-cuda:
     needs:
     - docker-python-base