Try newer cuda version

.github/workflows/cuda_ci.yaml

@@ -10,7 +10,7 @@ on: [push]
 jobs:
   test:
 
-    runs-on: ubuntu-latest
+    runs-on: ubuntu-20.04
     strategy:
       matrix:
         python-version: [3.8]
@@ -28,7 +28,7 @@ jobs:
       uses: Jimver/[email protected]
       id: cuda-toolkit
       with:
-        cuda: '11.2.2'
+        cuda: '12.5.0'
     - run: |
         echo "Installed cuda version is: ${{steps.cuda-toolkit.outputs.cuda}}"
         echo "Cuda install location: ${{steps.cuda-toolkit.outputs.CUDA_PATH}}"

azure-pipelines.yml

@@ -27,7 +27,7 @@ steps:
     wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb
     sudo dpkg -i cuda-keyring_1.1-1_all.deb
     sudo apt-get update
-    sudo apt-get -y install cuda-toolkit=11.2.2-1
+    sudo apt-get -y install cuda-toolkit
   displayName: Install nvcc
 
 - bash: conda env create --file environment.yml