Update unit_test_cpu.yaml with cpu nightly (#396)

pytorch · Jun 13, 2024 · a4cd9ba · a4cd9ba
1 parent e17e3b8
commit a4cd9ba
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/unit_test_cpu.yaml b/.github/workflows/unit_test_cpu.yaml
@@ -24,6 +24,6 @@ jobs:
 
         pip config --user set global.progress_bar off
 
-        pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu121
+        pip install --force-reinstall --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu
         pip install --pre torchdata --index-url https://download.pytorch.org/whl/nightly
         pytest test --cov=. --cov-report=xml --durations=20 -vv