- 确保已经安装完NVIDIA驱动,可利用下面的命令来验证
nvidia-smi
输出如下,代表NVIDIA驱动安装成功:
- 下载并安装CUDA
官网下载对应版本:CUDA Toolkit 11.7 Update 1 Downloads | NVIDIA Developer
- 配置环境变量
# 打开用户配置文件
vim ~/.bashrc
#添加如下代码,其中cuda-11.7修改为自己所安装的文件夹
export PATH=/usr/local/cuda-11.7/bin:$PATH
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.7/lib64
#加载环境变量,使其立即生效
source ~/.bashrc
- 验证CUDA是否安装成功
nvcc -V
- 安装CUDNN
- 下载CUDNN
cd ~/下载
tar -xvf cudnn-linux-x86_64-8.5.0.96_cuda11-archive
sudo cp include/cudnn*.h /usr/local/cuda-11.7/include/
sudo cp lib/libcudnn* /usr/local/cuda-11.7/lib64/
sudo chmod a+r /usr/local/cuda-11.7/include/cudnn*.h
sudo chmod a+r /usr/local/cuda-11.7/lib64/libcudnn*
- 验证CUDNN是否安装成功
cat /usr/local/cuda-11.7/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
若出现以下输出,代表安装成功。