Install nightly

pytorch · Dec 17, 2024 · f28fa86 · f28fa86
1 parent 6f42b08
commit f28fa86
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/.github/workflows/_linux-benchmark-cuda.yml b/.github/workflows/_linux-benchmark-cuda.yml
@@ -52,7 +52,8 @@ jobs:
         working-directory: benchmark
         run: |
           set -eux
-          ${CONDA_RUN} pip install torch torchvision torchaudio torchao
+          ${CONDA_RUN} pip3 install --pre torch torchvision torchaudio torchao \
+            --index-url https://download.pytorch.org/whl/nightly/cu124
 
       - name: Install benchmark
         shell: bash