Win10 安装Tensorflow-GPU版本

参考文章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
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值