文章目录
前言
强烈建议去官网查看相关的安装和使用方法
前置条件:
wsl2的ubuntu已安装cuda 13以上
步骤
- 启用网络存储库,并安装密钥包,使用如下命令:
cpp
$ wget https://developer.download.nvidia.com/compute/cuda/repos/<distro>/<arch>/cuda-keyring_1.1-1_all.deb
# sudo dpkg -i cuda-keyring_1.1-1_all.deb
# dpkg -i cuda-keyring_1.1-1_all.deb
注意 :第一行命令的<distro>和<arch>替换为你本人电脑的参数,如我使用的是ubuntu20.04和x86_64,那么替换后的完整命令为:
cpp
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb
# sudo dpkg -i cuda-keyring_1.1-1_all.deb
# dpkg -i cuda-keyring_1.1-1_all.deb
- 更新仓库:
cpp
sudo apt-get update
- 安装cuDNN包:
如果wsl2的cuda为12.xx请输入:
cpp
sudo apt-get -y install cudnn9-cuda-12
如果wsl2的cuda为12.xx请输入:
cpp
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:
