MAINT: upgrade cache and publish workflows with CUDANN upgrade

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: