我是靠谱客的博主 耍酷雪碧,这篇文章主要介绍ubuntu上安装cudnn5.1详细过程及多个cuda切换方法【食用就酱】一、cudnn5.1的安装二、查看cuDNN版本命令三、多个cuda的切换方法,现在分享给大家,希望可以做个参考。

本次记录在Ubuntu系统上复现代码时cuda以及cudnn的使用,期间参考了某些博主的文章,在这一如既往地对他们表示我由衷的感谢。另外,本人小小白一个,文中若有错误还望各位不吝赐教!

一、cudnn5.1的安装

首先下载安装文件,官网没找到,找到的一个百度云链接。
①下载好之后cd到相应文件夹

tar -xvf cudnn-8.0-linux-x64-v5.1.tgz -解压

③解压后得到一个 cudn 文件夹,该文件夹有include 和 lib64 两个文件夹,命令行进入 cudn/include 路径下,进行以下操作:

复制代码
1
2
3
cd cuda/include sudo cp cudnn.h /usr/local/cuda/include #复制头文件

④再进入lib64目录下的动态文件进行复制和链接:

复制代码
1
2
3
4
5
6
7
8
cd .. cd lib64 sudo cp lib* /usr/local/cuda/lib64/ #复制动态链接库 cd /usr/local/cuda/lib64/ sudo rm -rf libcudnn.so libcudnn.so.5 #删除原有动态文件 sudo ln -s libcudnn.so.5.0.5 libcudnn.so.5 #生成软衔接 sudo ln -s libcudnn.so.5 libcudnn.so #生成软链接

安装完成后可用 nvcc -V 命令验证是否安装成功,若出现版本等信息则表示安装成功。

二、查看cuDNN版本命令

复制代码
1
2
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

三、多个cuda的切换方法

复制代码
1
2
3
4
5
6
7
8
cd /usr/local/ -在/usr/local/目录下查看自己安装的cuda版本 stat cuda -查看目前指向哪个cuda sudo rm -rf cuda -删除软连接 sudo ln -s /usr/local/cuda-9.1 /usr/local/cuda -建立新的软连接

OK,本教程到此结束!下期再见!

最后

以上就是耍酷雪碧最近收集整理的关于ubuntu上安装cudnn5.1详细过程及多个cuda切换方法【食用就酱】一、cudnn5.1的安装二、查看cuDNN版本命令三、多个cuda的切换方法的全部内容,更多相关ubuntu上安装cudnn5.1详细过程及多个cuda切换方法【食用就酱】一、cudnn5.1内容请搜索靠谱客的其他文章。

本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
点赞(74)

评论列表共有 0 条评论

立即
投稿
返回
顶部