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