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