update fix

PennyLaneAI · Oct 10, 2023 · 643d9b0 · 643d9b0
1 parent 36ea421
commit 643d9b0
Showing 1 changed file with 5 additions and 7 deletions.
diff --git a/.github/workflows/test_gpu_cu11.yml b/.github/workflows/test_gpu_cu11.yml
@@ -9,8 +9,6 @@ env:
   CI_CUDA_ARCH: 86
   COVERAGE_FLAGS: "--cov=pennylane_lightning --cov-report=term-missing --cov-report=xml:./coverage.xml --no-flaky-report -p no:warnings --tb=native"
   GCC_VERSION: 11
-  CUDA_VERSION: 11.8
-  CUDA_MAJOR_VERSION: 11
   TF_VERSION: 2.10.0
   TORCH_VERSION: 1.11.0+cpu
 
@@ -36,7 +34,7 @@ jobs:
         run: |
           source /etc/profile.d/modules.sh
           module use /opt/modules
-          module load cuda/$CUDA_VERSION
+          module load cuda/11.8
           echo "${PATH}" >> $GITHUB_PATH
           echo "LD_LIBRARY_PATH=${LD_LIBRARY_PATH}" >> $GITHUB_ENV
           nvcc --version
@@ -57,7 +55,7 @@ jobs:
         run: |
           source /etc/profile.d/modules.sh
           module use /opt/modules
-          module load cuda/$CUDA_VERSION
+          module load cuda/11.8
           echo "${PATH}" >> $GITHUB_PATH
           echo "LD_LIBRARY_PATH=${LD_LIBRARY_PATH}" >> $GITHUB_ENV
           nvcc --version
@@ -105,7 +103,7 @@ jobs:
 
       - name: Install required packages
         run: |
-          python -m pip install ninja cmake custatevec-cu$CUDA_MAJOR_VERSION 
+          python -m pip install ninja cmake custatevec-cu11
       
       - name: Install Latest PennyLane
         if: inputs.pennylane-version == 'latest'
@@ -160,7 +158,7 @@ jobs:
         run: |
           source /etc/profile.d/modules.sh
           module use /opt/modules
-          module load cuda/$CUDA_VERSION
+          module load cuda/11.8
           echo "${PATH}" >> $GITHUB_PATH
           echo "LD_LIBRARY_PATH=${LD_LIBRARY_PATH}" >> $GITHUB_ENV
           nvcc --version
@@ -214,7 +212,7 @@ jobs:
       - name: Install required packages
         run: |
           python -m pip install pip~=22.0
-          python -m pip install ninja cmake custatevec-cu$CUDA_MAJOR_VERSION pytest pytest-mock flaky pytest-cov
+          python -m pip install ninja cmake custatevec-cu11 pytest pytest-mock flaky pytest-cov
 
       - name: Build and install package
         env: