为WSL2 ubuntu20.04安装cuDNN 9.2

前言

强烈建议去官网查看相关的安装和使用方法

前置条件:
wsl2的ubuntu已安装cuda 13以上

步骤

1. 启用网络存储库,并安装密钥包,使用如下命令:

$ wget https://developer.download.nvidia.com/compute/cuda/repos/<distro>/<arch>/cuda-keyring_1.1-1_all.deb#sudodpkg-i cuda-keyring_1.1-1_all.deb#dpkg-i cuda-keyring_1.1-1_all.deb

注意:第一行命令的<distro>和<arch>替换为你本人电脑的参数,如我使用的是ubuntu20.04和x86_64,那么替换后的完整命令为:

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb#sudodpkg-i cuda-keyring_1.1-1_all.deb#dpkg-i cuda-keyring_1.1-1_all.deb

2. 更新仓库:

sudo apt-get update

3. 安装cuDNN包:
   如果wsl2的cuda为12.xx请输入:

sudo apt-get-y install cudnn9-cuda-12

如果wsl2的cuda为12.xx请输入:

sudo apt-get-y install cudnn9-cuda-13

如果需要额外的jit,对于cuda12的输入:
sudo apt-get -y install cudnn9-jit-cuda-12
对于cuda13的输入:
sudo apt-get -y install cudnn9-jit-cuda-13

然后等待安装成功后即可成功运行cuDNN库的matmul: