.

Author: ahmadsharif1

.github/workflows/docs.yaml

@@ -30,8 +30,8 @@ jobs:
         run: |
           # 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/cpu
-          conda install "ffmpeg=7.0.1" pkg-config -c conda-forge
+          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
           ffmpeg -version
       - name: Build and install torchcodec
         run: |