关于已经安装cuda但是tensorflow仍然使用cpu加速的问题

平台:ubuntu16.04 LTS

环境:cuda9.1 tensorflow-gpu1.4 nvidia390.25 anconda3.0

问题:cuda配置成功,tensorflow也能够code,但是使用的却是cpu加速,

log:Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX AVX2. 

解决方法:使用conda list 检查已经安装的包,发现除了tensorflow-gpu之外,还有tensorflow 1.5。猜测可能是因为使用了cpu版本的tensorflow。于是卸载tensorflow,但是并没有起到效果。

尝试:conda uninstall protobuf,卸载全部tensorflow相关包,然后conda install tensorflow-gpu。随后编译提示切换设备为gpu0,成功使用gpu加速。

提示:keras和tensorflow不能共存。在conda install keras的时候会一起安装tensorflow,可能会导致上述问题。

参考:https://stackoverflow.com/questions/42326748/tensorflow-on-gpu-no-known-devices-despite-cudas-devicequery-returning-a-pas

评论 11
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值