win10环境下tensorflow-gpu安装

  1. 云栖社区>
  2. 博客>
  3. 正文

win10环境下tensorflow-gpu安装

ayew 2017-11-25 20:37:00 浏览449
展开阅读全文

 

 

 

 

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

pip install tensorflow-gpu

 

然后下载安装cuda8.0

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

注意:只能用8.0

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

然后:下载cudnn

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

 

要注册才能下载,这又是一个坑,选定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

没有报错就成功了。

 

网友评论

登录后评论
0/500
评论
ayew
+ 关注