Install TorchVision from nightly

pytorch · Nov 13, 2024 · 6cc503c · 6cc503c
1 parent ea90821
commit 6cc503c
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/.github/workflows/benchmarks.yaml b/.github/workflows/benchmarks.yaml
@@ -36,14 +36,14 @@ jobs:
       - name: Install FFmpeg, PyTorch and TorchAudio
         run: |
           conda install "ffmpeg=6" pkg-config -c conda-forge
-          python -m pip install --pre torch torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu
+          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
-        run: |
-          git clone https://github.com/pytorch/vision.git
-          cd vision
-          python setup.py install
+          #- name: Download, build and install TorchVision
+          #  run: |
+          #    git clone https://github.com/pytorch/vision.git
+          #    cd vision
+          #    python setup.py install
 
       - name: Install all other dependencies
         run: |