caffe 测试模型 Cannot use GPU in CPU-only Caffe : check mode.

在使用enet移植到win平台的CPU环境中,测试模型时遇到无法使用GPU的问题。解决方法为将Caffe模式从GPU更改为CPU。此更改确保了在没有GPU支持的环境下,模型测试能够正常进行。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

在使用enet移植到win平台cpu环境中时,测试模型时,出现

F0510 11:30:22.169340 13044 split_layer.cpp:53] Cannot use GPU in CPU-only Caffe : check mode.

解决方法
将测试源码中

Caffe::set_mode(Caffe::GPU);

改为

Caffe::set_mode(Caffe::CPU);

即可

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值