- Ubuntu16.04
python examples/ssd/ssd_pascal.py
利用Caffe执行模型训练时遇到未配置好CPU的问题:
Using GPUs 0, 1, 2,3
Cannot 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,以此类推