当前位置:首页 > 操作系统 > Windows

Windows7下安装Tensorflow GPU碰到的坑

经过N天摸索,总结如下:

1)tensorflow安装包必须用fo40225大神的,默认的pip的安装包不能工作,会报告numpy的问题

   估计问题的原因是CUDA/CUDNN的版本问题,不知道PIP默认的TENSORFLOW安装包用的什么版本。

   fo40225在github提供了CUDA/CUDNN,还有CPU支持的AVX2/SSE2指令的组合包,还有PYTHON版本。

   非常方便,按照组合准备环境就好。

   https://github.com/fo40225/tensorflow-windows-wheel

 

2)Anaconda的版本没有具体的要求,反正可以方便的在创建虚拟环境的时候指定python版本。

 

3)第一次启动tensorflow session花了8分钟的时间,第二次就好了,原因未知。

2019-02-24 20:22:38.253600: I c:usersusersourcerepostensorflowtensorflowc
orecommon_runtimegpugpu_device.cc:1484] Adding visible gpu devices: 0
2019-02-24 20:30:08.996800: I c:usersusersourcerepostensorflowtensorflowc
orecommon_runtimegpugpu_device.cc:965] Device interconnect StreamExecutor wit
h strength 1 edge matrix:

 


【说明】本文章由站长整理发布,文章内容不代表本站观点,如文中有侵权行为,请与本站客服联系(QQ:254677821)!

相关教程推荐

其他课程推荐