.

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 nvidia/label/cuda-12.4.0::cuda-toolkit nvidia/label/cuda-12.4.0::cuda-runtime
+          conda install nvidia/label/cuda-12.4.0::cuda-toolkit nvidia/label/cuda-12.4.0::cuda-runtime nvidia/label/cuda-12.4.0::cuda-cudart nvidia/label/cuda-12.4.0::cuda-nvrtc-dev
           conda install "ffmpeg=7.0.1" pkg-config cmake -c conda-forge
           ffmpeg -version
       - name: Build and install torchcodec