chnage build_and_test.yml

pytorch · May 17, 2024 · 792fd61 · 792fd61
1 parent 9fdf3ec
commit 792fd61
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/.github/workflows/build_and_test.yml b/.github/workflows/build_and_test.yml
@@ -79,6 +79,7 @@ jobs:
       timeout-minutes: 300
       collect-coverage: false
       install-cuda-plugin: true
+      require-pytorch-cuda: true
     secrets:
       gcloud-service-key: ${{ secrets.GCLOUD_SERVICE_KEY }}