您的位置:首页 > 其它

win10下安装TensorFlow

2017-10-28 09:59 246 查看
1. win10 下安装TensorFlow-CPU版本:
http://blog.csdn.net/u011159042/article/details/68065289
比较简单

2. win10 安装TensorFlow-gpu版本:

需要检查显卡能否安装;
http://blog.csdn.net/u010099080/article/details/53418159
安装时需要安装CUDA和cudnn

(1)下载CUDA

在官网上下了多次,都是发生错误,1.3G,下了四次都不行,下面链接里有
https://pan.baidu.com/share/init?surl=miBc2VY,提取码5aoc,可以直接用。
(2)安装CUDA

下载后,解压,按照提示进行安装即可。

(3)下载cuDNN

5.1版本,上面百度网盘里有;

但是我的出现了

ImportError: DLL load failed: 找不到指定的模块。
 和 
ImportError:
No module named '_pywrap_tensorflow_internal'
错误.如果在 import tensorflow
 的时候这两个问题同时出现,那么很有可能是你的 cuda 和 cudnn 版本有问题.
需要下载版本cudnn-6:

官网下载,需要填一堆问卷;

用下面百度网盘链接提取:
https://pan.baidu.com/s/1o8mc4YI


(3)安装cudnn-6
参考链接:http://blog.csdn.net/tongxxhaha/article/details/75752047

将包内三个文件复制到cuda相应位置,如图示。



 


同时还需将bin目录、lib\64目录添加到环境变量path中,否则后面调用时会报错。



测试,成功!
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: