.

Author: ahmadsharif1

.github/workflows/linux_cuda_wheel.yaml

@@ -108,7 +108,7 @@ jobs:
 
       - name: Install cuda runtime dependencies
         run: |
-          conda install --yes nvidia::libnpp nvidia::cuda-nvrtc nvidia::cuda-toolkit nvidia::cuda-cudart
+          conda install --yes nvidia::libnpp= nvidia::cuda-nvrtc nvidia::cuda-toolkit nvidia::cuda-cudart nvidia::cuda-driver-dev
       - name: Install test dependencies
         run: |
           python -m pip install --pre torchvision --index-url https://download.pytorch.org/whl/nightly/cpu
@@ -133,6 +133,9 @@ jobs:
         run: |
           find $CONDA_PREFIX -name libtorchcodec\*.so | xargs ldd
           find $CONDA_PREFIX -name libcuda\*.so\*
+          echo $CONDA_PREFIX
+          find / -name libcuda.so
+          find / -name libcuda.so.1
           python test/decoders/manual_smoke_test.py
       - name: Run Python tests
         run: |