Win下搭建tensorflow的GPU版本

1、下载python3.5.2版本并安装(必须是3.5版本,而且3.5后不带字母的版本,本人亲自踩坑!)

2http://www.lfd.uci.edu/~gohlke/pythonlibs/#numpy在该地址下载tensorflowGPU版本,,以及tensorflow的依赖protubuf,还有其它比如numpyscipy等,cudacuda8.044版本

3、安装cuda8.0,如果有报警信息也是可以装的(如截图),(印象中我犹豫了很久,最后是抱着试一下的态度装的),

安装完后看环境变量里有以下两项就表明cuda安装成功!

4、下载与cuda相匹配的cudnn,解压后(如下图)在环境变量的Path添加解压后的cudnnbin目录位置变量

5、到这里基本已经完成tensorflow的安装配置了,如果想验证一下cudnn安装成功,可以建立一个VS的控制台项目,配置平台为x64release版本(不是debug哦),然后在项目属性—>VC++目录—>包含目录加:D:\cudnn-8.0-windows10-x64-v5.1\cuda\include(你cudnninclude目录),库目录加:D:\cudnn-8.0-windows10-x64-v5.1\cuda\lib\x64(你cudnnlib目录),链接器—>附加依赖项添加:cudnn.lib。添加如下代码:

效果输出如果是:CUDNN_STATUS_SUCCESS,表明CUDNN安装成功!

6、测试一个tensorflow版本的mnist程序,在cmd命令里运行,有输出结果如下:

那恭喜你,表明GPU版本的tensorflow安装成功!

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值