Install PyTorch, TorchAudio and TorchVision from conda

pytorch · Nov 13, 2024 · 0f0412b · 0f0412b
1 parent 9e59bab
commit 0f0412b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/.github/workflows/benchmarks.yaml b/.github/workflows/benchmarks.yaml
@@ -36,7 +36,8 @@ jobs:
       - name: Install FFmpeg, PyTorch and TorchAudio
         run: |
           conda install "ffmpeg=6" pkg-config -c conda-forge
-          python -m pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu
+          conda install pytorch torchvision torchaudio cpuonly -c pytorch-nightly
+          #python -m pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu
           ffmpeg -version
 
           #- name: Download, build and install TorchVision