(Ubuntu安装四)Ubuntu16.04安装cuDNN


参考链接: https://blog.csdn.net/qq_40936141/article/details/81198425#commentBox

一、官网注册账号并下载cuDNN

1、版本选择

cuDNN版本千万千万别选错!!!

我这里选择 cuDNN v7.05[Dec 5,2017],for CUDA9.0
我安裝的是 cuDNN v7.0.5 Library for Linux
在这里插入图片描述
如上图所示,CUDA-9.0版本的必须和cuDNN的7.0的版本对应起来。cuDNN版本不能超过7.4,否则后面安装tensorflow-gpu会报错!!!!!

2、具体下载步骤

在这里插入图片描述在这里插入图片描述在这里插入图片描述在这里插入图片描述在这里插入图片描述
下载完成。
在这里插入图片描述

二、解压文件,复制到本地路径

在“下载”文件夹中打开终端,依次执行以下命令:

tar -zxvf cudnn-9.0-linux-x64-v7.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda-9.0/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-9.0/lib64/

在这里插入图片描述

三、进入本地的cuda路径的lib64文件夹

本地的cuda路径的lib64文件夹会有刚刚解压的文件,然后在终端打开依次执行以下命令:

cd /usr/local/cuda/lib64
 
sudo chmod +r libcudnn.so.7.0.5
 
sudo ln -sf libcudnn.so.7.0.5 libcudnn.so.7
 
sudo ln -sf libcudnn.so.7 libcudnn.so
 
sudo ldconfig

接下来在终端输入:ll
安装成功会出现以下信息。
在这里插入图片描述

四、整体流程截图

在这里插入图片描述在这里插入图片描述

五、卸载cuDNN

依次输入下面两条命令进行卸载:

sudo rm -rf /usr/local/cuda/include/cudnn.h
sudo rm -rf /usr/local/cuda/lib64/libcudnn
发布了28 篇原创文章 · 获赞 5 · 访问量 1855

猜你喜欢

转载自blog.csdn.net/elegantoo/article/details/103891723