diff --git a/.github/workflows/cache.yml b/.github/workflows/cache.yml index b1111b7..89003fc 100644 --- a/.github/workflows/cache.yml +++ b/.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: diff --git a/.github/workflows/publish.yml b/.github/workflows/publish.yml index c72fff2..8e195c3 100644 --- a/.github/workflows/publish.yml +++ b/.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: