CUDA 安装教程

首先, 打开Nvidia CUDA Toolkit Archive 的网站, 选择需要安装的版本, 我这里以11.8 为例

依次选择 Linux x86_64 WSL-Ubuntu 2.0 runfile

复制下面的指令

wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run
sudo sh cuda_11.8.0_520.61.05_linux.run

输入安装指令

发现要安装gcc

安装gcc

apt install gcc -y

再次运行安装指令,接受协议

选择并安装

看到日志显示安装成功

输入nvcc -V 发现无法调用

查看安装好的cuda版本

ls -l /usr/local

发现cuda安装, 并已经指向了正确的位置, 现在需要修改.bashrc 文件, 让系统能够调用

nano .bashrc

滑倒最底下, 输入以下内容并保存

export PATH=/usr/local/cuda-11.8/bin${PATH:+:${PATH}}

更新 .bashrc

source ~/.bashrc

使用 nvcc -V 检查, 可以看到已经正确安装并调用了11.8的cuda

cuDNN 安装教程

打开 cuDNN Archive 网站, 选择需要安装的cuDNN和对应的cuda版本

选择对应自己版本的cuDNN

登录账户并获取下载到/root 目录, 我这里的文件名是cudnn-linux-x86_64-8.9.7.29_cuda11-archive.tar.xz

复制到剪切板

tar -xvf cudnn-linux-x86_64-8.9.7.29_cuda11-archive.tar.xz
sudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda/include
sudo cp -P cudnn-*-archive/lib/libcudnn* /usr/local/cuda/lib64 
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

验证cuDNN版本

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

看到正确输出了cuDNN版本则是安装成功