.

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 nvidia::cuda-driver-dev
+          conda install --yes nvidia::libnpp nvidia::cuda-nvrtc=12.4 nvidia::cuda-toolkit=12.4 nvidia::cuda-cudart=12.4 nvidia::cuda-driver-dev=12.4
       - name: Install test dependencies
         run: |
           python -m pip install --pre torchvision --index-url https://download.pytorch.org/whl/nightly/cpu