win10环境下tensorflow-gpu安装

win10环境下tensorflow-gpu安装

踩了很多坑,浪费了一下午时间,在此记录一下。

pip install tensorflow-gpu

然后下载安装cuda8.0

https://developer.nvidia.com/cuda-downloads

win10环境下tensorflow-gpu安装

注意:只能用8.0

我用9.0下载了好久,程序提示我装8.0

然后:下载cudnn

https://developer.nvidia.com/rdp/cudnn-download

win10环境下tensorflow-gpu安装

要注册才能下载,这又是一个坑,选定cudnn6.0版本

其中,cudnn的安装方法是将cudnn下面的三个文件夹(bin,include, lib)复制到cuda 8.0文件夹。

装完以后运行,出现以下错误:

ImportError: No module named '_pywrap_tensorflow_internal'

ImportError: DLL load failed: The specified module could not be found.

解决的办法是把cudnn64_6.dll改名为cudnn64_5.dll。

然后

import tensorflow

没有报错就成功了。

上一篇:HTML 练习绑定onclick事件


下一篇:java多线程制作计时器