ubuntu16.04更改CUDNN 6.0 版本
在 ubuntu16.04 上安装tensorflow-gpu发现没有对应版本。查看之后,我的电脑是cuda 8.0,cudnn 7版本
所以需要更换cudnn版本为6.0,才可以安装tensorflow-gpu
一、下载linux版的CUDNN 6.0
官网下载:https://developer.nvidia.com/cuDNN,需要邮箱注册。
二、安装
1)解压
到下载的目录下:
tar zxvf cudnn-8.0-linux-x64-v6.0.tgz
2) 删除原来的CUDNN文件
sudo rm -rf /usr/local/cuda/include/cudnn.h
sudo rm -rf /usr/local/cuda/lib64/libcudnn*
3)cd 到之前解压的cuda文件夹下
sudo cp include/cudnn.h /usr/local/cuda/include/
sudo cp lib64/lib* /usr/local/cuda/lib64/
4)cd到/usr/local/cuda/lib64/文件夹下,建立软链接
cd /usr/local/cuda/lib64/
sudo chmod +r libcudnn.so.6.0.21
sudo ln -sf libcudnn.so.6.0.21 libcudnn.so.6
sudo ln -sf libcudnn.so.6 libcudnn.so
sudo ldconfig
5)查看版本信息
cuda版本
cat /usr/local/cuda/version.txt
cudnn版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2