网上有一大堆的安装方法
这里讲一下我碰到的一些问题。
安装cuda
1.官网下载cuda
这里有教你怎么安装:sudo sh cuda_10.1.105._418_39_linux.run
当然下载之前要对应cuDNN版本
其中cuDNN要注册。
2.装cuda
accept #同意安装
n #不安装Driver,因为已安装最新驱动
y #安装CUDA Toolkit
<Enter> #安装到默认目录
y #创建安装目录的软链接 即在/usr/local/下会有一个cuda文件夹,它是有cuda-10.1 软链接过去的,用于可以随时切换不同版本的cuda(前提是你装了不同版本的cuda)
n #不复制Samples
搞定之后(如果报Error:unsupported compiler:7.4.0的错,把gcc版本降低到6就可以了)
在~/.bashrc中的添加环境变量如下
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
添加完后记得输入命令:source ~/.bashrc
然后输入:
nvcc -V ####查看CUDA版本
其中环境变量那里每篇博文都不怎么一样,不知道会不会有影响。
这里还有关于cuda9.0和8.0的切换,利用到软链接,可供参考:https://blog.csdn.net/u010821666/article/details/79957071
安装cuDNN
官网上下载cuDNN压缩包,解压到cuda文件夹(就是软链接那个文件夹)就行了。具体代码:(参考https://www.cnblogs.com/alexanderkun/p/6908191.html)
1)cd进入cudnn解压之后的include目录,在命令行进行如下操作:
sudo cp cudnn.h /usr/local/cuda/include/ #复制头文件
2)再cd进入解压之后lib64目录下的动态文件进行复制和链接:
sudo cp lib* /usr/local/cuda/lib64/ #复制动态链接库
接下来的动作我没做(因为/usr/local/cuda/lib64/里面没有 libcudnn.so 和libcudnn.so.5 两个文件)
cd /usr/local/cuda/lib64/
sudo rm -rf libcudnn.so libcudnn.so.5 #删除原有动态文件
sudo ln -s libcudnn.so.5.1.5 libcudnn.so.5 #生成软衔接
sudo ln -s libcudnn.so.5 libcudnn.so #生成软链接
然后就完了,装好了
本文相关参考博文:1.https://blog.csdn.net/qq_36122764/article/details/82623089