MAINT: upgrade cache and publish workflows with CUDANN upgrade (#220)

* misc

* @mmcky edits

* enable dropdown

* upgrade CUDANN to 9.10.2

* MAINT: upgrade cache and publish workflows with CUDANN upgrade

---------

Co-authored-by: John Stachurski <john.stachurski@gmail.com>

Author: mmcky

.github/workflows/cache.yml

@@ -11,6 +11,13 @@ jobs:
       - uses: actions/checkout@v4
         with:
           ref: ${{ github.event.pull_request.head.sha }}
+      - name: Upgrade 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 cudnn-cuda-12
       - name: Setup Anaconda
         uses: conda-incubator/setup-miniconda@v3
         with:

.github/workflows/publish.yml

@@ -10,6 +10,13 @@ jobs:
     steps:
       - name: Checkout
         uses: actions/checkout@v4
+      - name: Upgrade 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 cudnn-cuda-12
       - name: Setup Anaconda
         uses: conda-incubator/setup-miniconda@v3
         with: