tensorflow使用cpu不用gpu的问题

在已安装cuda和tensorflow-gpu的环境中,若运行模型时CPU资源满载而GPU闲置,可能原因是tensorflow-gpu版本低于tensorflow版本或cuda版本不匹配。解决方法包括:检查tensorflow-gpu与tensorflow的版本兼容性,通过nvidia-smi检查GPU使用情况,以及核对cuda和cudnn版本。确保所有组件版本匹配并适当升级,以便利用GPU加速计算。
摘要由CSDN通过智能技术生成

前提:cuda版本和cudnn版本已安装好,版本对应没问题;

问题:在安装好cuda、tensorflow、tensorflow-gpu的环境下,调用tensorflow跑模型,发现cpu资源占满,而gpu处于空闲状态。

导致上述问题的原因可能有两个:

(1)tensorflow-gpu的版本低于tensorflow的版本
(2)cuda版本与tensorflow-gpu版本对应不上

对于第一个原因:tensorflow-gpu的版本低于tensorflow的版本,有以下检查方法:

(1)跑模型,然后bash环境输入nvidia-smi直接查看是否正使用gpu(也能查看CUDA版本)

(2)在python环境下打印设备信息

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

        1)假如tensorflow-gpu版本低于tensorflow版本时,只会输出以下设备全是cpu的信息:

                                         

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值