Ubantu系统和显卡驱动是小伙伴帮我安装的,cuda10.0和cudnn是我自己安装的,所以在此只说明cuda10.0和cudnn的安装过程~
一、安装cuda
在官网(链接1)下载安装文件:
选择完对应版本之后,点击红框下载,第二个需不需要下载我也不太清楚,我没有下载,暂时没啥影响。然后在下载好的文件路径下输入安装命令:
sudo sh cuda_10.0.130_410.48_linux.run
接下拉按照步骤选择: accept --> n --> y --> y --> y --> ... ,还有几个enter。
安装完成后,设置环境变量。先在终端输入命令来打开文件:
sudo gedit ~/.bashrc
然后在文件末尾添加:
export CUDA_HOME=/usr/local/cuda-10.0
export LD_LIBRARY_PATH=/usr/lcoal/cuda-10.0/lib64:$LD_LIBRARY_PATH
export PATH=/usr/local/cuda-10.0/bin:$PATH
保存退出后,在终端运行:
source ~/.bashrc
应该已经安装好啦,检查一下是否安装成功:
nvcc -V
还不太够,再检查一下:
cd /usr/local/cuda-10.0/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
据说出现 Result = PASS 就没问题了,如果出现 Result = FAIL 就有问题了,可能是驱动版本和cuda版本不匹配,此处参考链接2。
二、安装cudnn
首先,去官网(链接3)下载文件,官网需要注册账号。
解压下载好的文件,打开后可以看到cuda文件夹,
在终端打开解压好的文件夹的目录下,以此输入以下四条指令:
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
最后输入以下指令查看一下cudnn的版本,顺便检查一下有没有安装成功~
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
结果如下,说明安装成功啦~ 我的cudnn版本是7.6.4,没毛病~
大功告成~! 记录一下安装过程,供大家参考~