.

pytorch · Nov 7, 2024 · 1507669 · 1507669
1 parent 89c380e
commit 1507669
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/docs.yaml b/.github/workflows/docs.yaml
@@ -31,7 +31,7 @@ jobs:
           pip3 install --pre torch torchvision torchaudio torchcodec --index-url https://download.pytorch.org/whl/nightly/cu124
       - name: Install FFMPEG and other deps
         run: |
-          conda install cuda-nvrtc=12.4 libnpp -c nvidia
+          conda install cuda-nvrtc=12.4 libnpp cuda-nvcc=12.4 cuda-cudart=12.4 -c nvidia
           conda install ffmpeg=7 cmake pkg-config -c conda-forge
           ffmpeg -version
       - name: Build torchcodec