天天看點

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

繼續閱讀