.

Author: ahmadsharif1

.github/workflows/docs.yaml

@@ -31,7 +31,7 @@ jobs:
           # TODO: torchvision and torchaudio shouldn't be needed. They were only added
           #  to silence an error as seen in https://github.com/pytorch/torchcodec/issues/203
           python -m pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124
-          conda install "ffmpeg=7.0.1" pkg-config cmake -c conda-forge
+          conda install "ffmpeg=7.0.1" pkg-config cmake nvidia/label/cuda-12.4.0::cuda-toolkit -c conda-forge
           ffmpeg -version
       - name: Build and install torchcodec
         run: |