tensorflow 1.8 无法调用GPU的问题

代码:https://github.com/dennybritz/cnn-text-classification-tf

1) spyder情况下运行课程以上链接中的train.py,错误提示:InternalError (see above for traceback): CUB reduce errorinvalid device function。我之前用CPU版本的 tf 是可以运行的,跑了一个小时没完,我才切换到GPU的tf版本,然后问题来了...

2)cuda应该是安装正确的,9.0版本,win7系统,证据如下

3)cudnn 是7.x的版本,这个不知怎么检测是否安装争取,但应该没啥问题

4)tf-gpu安装的是1.8,py3.6版本

5)用官方的方法不显示 tf 是否调用的GPU,只显示个结果,如下图。但网上都说显示的。。。

https://blog.csdn.net/qq_33186949/article/details/79104659

6)后面我在jupter里面尝试了上面的代码,在日志里面显示如下,这应该是调用了GPU才对的

小弟是做天线的,目前就只有matlab处理简单数据的经验,周六加班周天搞了一天也没搞定,求大佬指教,感激不尽~~

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值