.

Author: ahmadsharif1

.github/workflows/linux_cuda_wheel.yaml

@@ -85,8 +85,7 @@ jobs:
         run: python -m pip install --upgrade pip
       - name: Install PyTorch
         run: |
-          cuda_version_without_periods=$(echo "${{ matrix.cuda-version }}" | sed 's/\.//g')
-          python -m pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cu$cuda_version_without_periods
+          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