几经周折后终于安装好了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包安装。