cuda+cuDNN

网上有一大堆的安装方法
这里讲一下我碰到的一些问题。
安装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

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值