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