Pin CUDA nightly to mitigate regression (pytorch#322)

Mitigates pytorch/pytorch#127924
dbyoung18 · Jun 4, 2024 · 0669d1d · 0669d1d
1 parent 9d1d8df
commit 0669d1d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/regression_test.yml b/.github/workflows/regression_test.yml
@@ -33,7 +33,7 @@ jobs:
             gpu-arch-version: "12.1"
           - name: CUDA Nightly 
             runs-on: linux.g5.12xlarge.nvidia.gpu
-            torch-spec: '--pre torch --index-url https://download.pytorch.org/whl/nightly/cu121'
+            torch-spec: '--pre torch==2.4.0.dev20240603+cu121 --index-url https://download.pytorch.org/whl/nightly/cu121'
             gpu-arch-type: "cuda"
             gpu-arch-version: "12.1"
           - name: CPU 2.2.2