续接前文:

卸载旧的cuda安装新的cuda【CUDA降级】

进入CUDNN的下载库中选择一个适合的版本:https://developer.nvidia.com/rdp/cudnn-archive

如我选择的是 cuDNN v7.1.3 Library for Linux 对应的CUDA版本是9.0。

这样下载的是一个tgz压缩文件,我们在下载之后进行解压缩。

tar -zxvf cudnn-9.0-linux-x64-v7.1.tgz

然后将其中的库文件内容进行复制即可:

sudo cp cuda/lib64/* /usr/local/cuda/lib64/
sudo cp cuda/include/* /usr/local/cuda/include/

#或者:这决定与你的环境变量如何书写,本人使用上面的
sudo cp cuda/lib64/* /usr/local/cuda-9.0/lib64/
sudo cp cuda/include/* /usr/local/cuda-9.0/include/

可以使用命令查看CUDNN版本号信息:

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
#define CUDNN_MAJOR 7
#define CUDNN_MINOR 1
#define CUDNN_PATCHLEVEL 3
--
#define CUDNN_VERSION    (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)

#include "driver_types.h"

即说明正确,尝试在python下import tensorflow没有报错即说明深度学习GPU部分安装完毕。


0 条评论

发表回复

Avatar placeholder

您的电子邮箱地址不会被公开。 必填项已用 * 标注