Skip to content

Commit 0f0412b

Browse files
committed
Install PyTorch, TorchAudio and TorchVision from conda
1 parent 9e59bab commit 0f0412b

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

.github/workflows/benchmarks.yaml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,8 @@ jobs:
3636
- name: Install FFmpeg, PyTorch and TorchAudio
3737
run: |
3838
conda install "ffmpeg=6" pkg-config -c conda-forge
39-
python -m pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu
39+
conda install pytorch torchvision torchaudio cpuonly -c pytorch-nightly
40+
#python -m pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu
4041
ffmpeg -version
4142
4243
#- name: Download, build and install TorchVision

0 commit comments

Comments
 (0)