您的位置:首页 > 其它

ubuntu安装cudnn

2019-08-04 16:31 309 查看
版权声明:本文为博主原创文章,遵循 CC 4.0 by-sa 版权协议,转载请附上原文出处链接和本声明。 本文链接:https://blog.csdn.net/public669/article/details/98470857

有些忙,这一段时间,博客就随便写写了~~~
默认cuda安装好了,这里就不多说了,我们从cuda的环境变量开始说起:
配置cuda环境变量:
打开终端,输入”gedit ~/.bashrc“,然后再弹出的编辑器的最后,添加如下两行

export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

测试CUDA
进入你的CUDA Example所在目录,默认是主目录,找到“NVIDIA_CUDA-10.0_Samples”。依次打开“1_Utilities”–>“deviceQuery”,然后重新打开一个终端输入:

# 使.bashrc的环境变量生效,重启之后或者打开新终端就不用再执行这一行了
source ~/.bashrc
# 修改文件权限
sudo chmod -R 777 *
# 编译
make
# 运行
./deviceQuery

出现”Result = PASS“字样时,说明安装成功了
安装CUDNN:
地址:https://developer.nvidia.com/rdp/cudnn-archive
选择符合自己cuda版本的
下载“cuDNN Library for Linux”那一个,
安装cudnn
安装过程实际上是把cudnn的头文件复制到CUDA的头文件目录里面去;把cuDNN的库复制到CUDA的库目录里面去。
首先需要将下载的cudnn解压,之后再执行如下命令:(大家也可以一个一个的移动到相应的cuda文件下)

# 复制cudnn头文件
sudo cp cuda/include/* /usr/local/cuda-10.0/include/
# 复制cudnn的库
sudo cp cuda/lib64/* /usr/local/cuda-10.0/lib64/
# 添加可执行权限
sudo chmod +x /usr/local/cuda-10.0/include/cudnn.h
sudo chmod +x /usr/local/cuda-10.0/lib64/libcudnn*

检验:
输入如下命令:

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

出现如下则证明安装成功:

内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: