Win10 安装Tensorflow-GPU版本

来源:互联网 发布:重庆时时彩开奖源码 编辑:程序博客网 时间:2024/05/13 00:35

参考文章1【TensorFlow】Windows10 64位下安装TensorFlow - 官方原生支持,文章2Anaconda多环境多版本python配置指导及官方安装文档

前几天google放出了支持win10 GPU的tensorflow 0.12版本,现在安装体验一下,主要参考了如上文章。

安装CUDA8.0和cuDNN5

从官方链接1和官方链接2即可下载,不过cuDNN下载需要注册开发者。之前我已经安装了vs2012。
安装参考文章1即可,需要注意的是,安装后如果要测试demo,需要注意c++工程需要先生成解决方案(编译链接)然后才可以进行调试执行。另外,多个项目中需要右键来确认选择启动项目。

配置及安装tensorflow

win10只支持python3.5,而我常用的是python2.7需要利用conda建立专门的环境,如下:

conda create -n tensorflow python=3.5

win10下切换环境,参考文章2

activate tensorflow
deactivate

在tensorflow环境下安装tensorflow-gpu,不要安装tensorflow(默认CPU版本)

pip install tensorflow-gpu
pip uninstall tensorflow-gpu

最终测试,参考文章1即可,如果正确使用GPU版本,将会在计算过程中输出可用的GPU信息。
以上。

0 0
原创粉丝点击