From 23078648611a2bc78a5ad84faebade90651501a0 Mon Sep 17 00:00:00 2001 From: Timur Daukaev Date: Tue, 26 Nov 2024 11:08:46 +0100 Subject: [PATCH] MP-1215 Use prebuilt pytorch for cpu images --- .github/workflows/build.yml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml index 812b2f2..852b31a 100644 --- 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