ubuntu中安装cuDNN

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/liuYinXinAll/article/details/79979436

几经周折后终于安装好了cuda,接下来根据tensorflow的要求,开始安装cudnn
官方教程地址

1. 下载cudnn

要注册一个nvidia的帐号才可以下载
下载地址
根据系统版本,选择相应的版本。我的是ubuntu,选择通过压缩包的形式安装
cudnn-9.0-linux-x64-v7.tgz

2. 安装

然后解压

tar -xvzf cudnn-9.1-linux-x64-v7.1.tgz 

会在当前目录下创建一个 cuda 的文件夹,包括以下文件。

cuda/include/cudnn.h
cuda/NVIDIA_SLA_cuDNN_Support.txt
cuda/lib64/libcudnn.so
cuda/lib64/libcudnn.so.7
cuda/lib64/libcudnn.so.7.1.2
cuda/lib64/libcudnn_static.a

然后把这些文件按照类别放到cuda中

sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo chmod a+r /usr/local/cuda/include/cudnn.h 
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

然后就好了。另外ubuntu中还支持通过deb包安装。

阅读更多
想对作者说点什么?

博主推荐

换一批

没有更多推荐了,返回首页