当安装好cuda后,由于使用tensorflow需要调用gpu,因而需要cudnn的支持,再此安装cudnn。
接下来安装CUDNN7.0.5
官网下载压缩包,然后解压,解压好了之后是一个叫cuda的文件夹。
把里面的lib库添加到之前cuda的目录下面,假设我现在在cuda(cudnn的解压包)文件夹下面
sudo cp ./include/cudnn.h /usr/local/cuda-9.0/include
sudo cp ./lib64/libcudnn* /usr/local/cuda-9.0/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda-9.0/lib64/libcudnn*
关于更新软链接,有的人做了有的人没有,貌似没更新只会有warning。
更新软链接(同样下面版本注意看清楚)
cd /usr/local/cuda/lib64/
sudo rm -rf libcudnn.so libcudnn.so.7
sudo chmod u=rwx,g=rx,o=rx libcudnn.so.7.0.5
sudo ln -s libcudnn.so.7.0.5 libcudnn.so.7
sudo ln -s libcudnn.so.7 libcudnn.so
sudo ldconfig
即可在tensorflow中运行gpu.