.

Author: ahmadsharif1

.github/workflows/linux_cuda_wheel.yaml

@@ -61,8 +61,9 @@ jobs:
       fail-fast: false
       matrix:
         python-version: ['3.9']
-        cuda-version: ['124']
-        ffmpeg-version-for-tests: ['4.4.2', '5.1.2', '6.1.1', '7.0.1']
+        cuda-version: ['12.4']
+        cuda-version-without-period: ${{ replace(matrix.cuda-version, ".", "")}}
+        ffmpeg-version-for-tests: ['5.1.2', '6.1.1', '7.0.1']
     if: ${{ always() }}
     needs: build
     steps:
@@ -81,7 +82,7 @@ jobs:
         run: python -m pip install --upgrade pip
       - name: Install PyTorch
         run: |
-          python -m pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu${{ matrix.cuda-version }}
+          python -m pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu${{ matrix.cuda-version-without-period }}
       - name: Install cuda dependencies
         run: |
           conda install --yes nvidia::libnpp