update CUDA=12.9

Author: mmcky

.github/workflows/ci.yml

@@ -16,6 +16,13 @@ jobs:
           python-version: "3.12"
           environment-file: environment.yml
           activate-environment: quantecon
+      - name: Install CUDA and CUDANN
+        shell: bash -l {0}
+        run: |
+          wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/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-12-9
       - name: Install jax (and install checks for GPU)
         shell: bash -l {0}
         run: |