安装完CUDA8以后,要安装对应的cuDNN。
环境:(环境不一致也可以参考一下,但是不能无脑参考了)
Ubantu 16.04
Cuda 8GA2(没有安装update)
1.下载cuDNN
tensorflow官方推荐cuDNN6,所以去下载6,去这个网址:
cuDNN点击下载,然后注册登录,服务器偶尔抽风注册不了。。这个我也没办法,去网盘里看有没有人分享把。(虽然官网写了下载7,但是你登录进去点下载会给你选择下载哪个版本的)
好了,下载下来,是一个solitairetheme8文件,把它拷进你的ubantu里。
2.解压
输入如下:
cp cudnn-8.0-linux-x64-v6.0.solitairetheme8 cudnn-6.0.tgz
tar -xvf cudnn-8.0-linux-x64-v5.1.tgz
3.安装
与其叫安装,不如叫复制。
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*