1. 安装NVIDIA驱动
首先检查系统推荐安装的版本
ubuntu-drivers devices
安装推荐版本驱动
sudo apt install nvidia-driver-XXX
安装完毕后重启,检查
nvidia-smi
2. 安装CUDA
从NVIDIA官网选择合适版本下载
推荐使用runfile安装
在安装过程中选择Continue,输入accept,使用空格去掉Driver,最后install安装
安装完毕之后修改路径文件
gedit ~/.bashrc
在最后加上以下两句,注意修改CUDA版本
export PATH=/usr/local/cuda-1X.X/bin${PATH:+:${PATH}}export LD_LIBRARY_PATH=/usr/local/cuda-1X.X/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
保存路径
source ~/.bashrc
查看CUDA版本以验证安装成功
nvcc -V
3. 安装CUDNN
从官网中下载合适版本,需要注册登录NVIDIA账号
下载完之后,解压文件夹,并复制文件(注意修改路径):
将cuda/include/cudnn.h文件复制到usr/local/cuda/include文件夹
sudo cp path/cuda/include/cudnn.h /usr/local/cuda/include
将cuda/lib64/下所有文件复制到/usr/local/cuda/lib64文件夹
sudo cp path/cuda/lib64/* /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.h | grep cudnnfind / -name cudnn_version.h 2>&1 | grep -v "Permission denied"
使用以下命令查看CUDNN版本(注意修改文件路径)
cat path/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
感谢以下作者:
@ LUKE [链接]
@ashome123 [链接]
@哈希Ma [链接]
@ eaxy_z [链接]