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,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，以此类推