300字范文,内容丰富有趣,生活中的好帮手!
300字范文 > CPU配置Caffe训练SSD遇到Cannot use GPU in CPU-only Caffe: check mode问题解决办法

CPU配置Caffe训练SSD遇到Cannot use GPU in CPU-only Caffe: check mode问题解决办法

时间:2020-07-31 09:10:31

相关推荐

CPU配置Caffe训练SSD遇到Cannot use GPU in CPU-only Caffe: check mode问题解决办法

Ubuntu16.04

python examples/ssd/ssd_pascal.py

利用Caffe执行模型训练时遇到未配置好CPU的问题:

Using GPUs 0, 1, 2,3Cannot use GPU in CPU-only Caffe: check mode.

解决办法:

打开 /home/xxx/caffe/examples/ssd/ssd_pascal.py文件,注释以下几行

# if num_gpus > 0:# batch_size_per_device = int(math.ceil(float(batch_size) / num_gpus))# iter_size = int(math.ceil(float(accum_batch_size) / (batch_size_per_device * num_gpus)))# solver_mode = P.Solver.GPU# device_id = int(gpulist[0])

保存后终端运行

python examples/ssd/ssd_pascal.py

如果出现问题cudasuccess ( 2vs0 ) 则说明你的显卡计算量有限,再次打开ssd_pascal.py文件,找到这一行

batch_size = 32

修改数字32,可以修改为16,8,甚至4,保存后终端运行

python examples/ssd/ssd_pascal.py

针对应用GPU配置不同显卡数的问题:

在ssd_pascal.py中找到这一行

gpus = "0,1,2,3"

如果服务器有一块显卡,则删去1,2,3,如果有两块显卡,删去2,3,以此类推

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。