1、下载CUDA9.0
打开下载地址:https://developer.nvidia.com/cuda-90-download-archive
2、下载后进入对应文件夹,输入sudo sh cuda_9.0.176_384.81_linux.run
然后一直Enter到100%,并输入accept
接着出现 Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 384.81?
一定要选择No,否则之前的驱动安装就白安了
3、添加环境变量
sudo gedit /etc/profile
打开“profile”文件,在末尾处添加(注意不要有空格,不然会报错):
export PATH=/usr/local/cuda-9.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64$LD_LIBRARY_PATH
重启电脑
4、输入nvcc -V,测试是否安装成功
5、安装CUDNN v7.05
下载地址:https://developer.nvidia.com/rdp/cudnn-archive#a-collapse705-9
6、下载后解压文件,解压之后,得到一个 cudn 文件夹,该文件夹下include 和 lib64 两个文件夹,命令行进入 cuda/include 路径下,然后进行以下操作:
cd cuda/include
sudo cp cudnn.h /usr/local/cuda/include #复制头文件
再进入lib64目录下的动态文件进行复制和链接:
cd ..
cd lib64
sudo cp lib* /usr/local/cuda/lib64/ #复制动态链接库
cd /usr/local/cuda/lib64/
sudo rm -rf libcudnn.so libcudnn.so.7 #删除原有动态文件
sudo ln -s libcudnn.so.7.0.5 libcudnn.so.7 #生成软衔接
sudo ln -s libcudnn.so.7 libcudnn.so #生成软链接
sudo ldconfig
验证CUDNN是否安装成功:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
最后推荐一个ubuntu下的截图软件,方便写博客使用:Shutter
参考地址:https://jingyan.baidu.com/article/4b52d702db5482fc5c774bf0.html