.

Author: ahmadsharif1

.github/workflows/linux_cuda_wheel.yaml

@@ -88,8 +88,7 @@ jobs:
           python -m pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu${{ env.cuda_version_without_periods }}
       - name: Install cuda dependencies
         run: |
-          conda install --yes nvidia::libnpp
-          conda install --yes nvidia::cuda-nvrtc
+          conda install --yes nvidia::libnpp nvidia::cuda-nvrtc nvidia::cuda-toolkit nvidia::cuda-cudart
       - name: Install torchcodec from the wheel
         run: |
           wheel_path=`find pytorch/torchcodec/dist -type f -name "*.whl"`