Fixed nightly version

pytorch · Jan 24, 2025 · cc06016 · cc06016
1 parent 644cd10
commit cc06016
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/float8_test.yml b/.github/workflows/float8_test.yml
@@ -30,7 +30,7 @@ jobs:
             gpu-arch-version: "12.4"
           - name: H100
             runs-on: linux.aws.h100
-            torch-spec: '--pre torch --index-url https://download.pytorch.org/whl/nightly/cu124'
+            torch-spec: '--pre torch==2.6.0 --index-url https://download.pytorch.org/whl/nightly/cu124'
             gpu-arch-type: "cuda"
             gpu-arch-version: "12.4"