要在Caffe中使用GPU加速訓練,首先需要確保你的計算機上安裝了CUDA和cuDNN。然后,在進行訓練之前,需要在配置文件中將solver_mode設置為GPU。接著,可以在訓練命令中加入-gpu參數來指定使用的GPU設備編號。最后,運行訓練命令時,Caffe會自動利用GPU加速訓練過程。
億速云公眾號
手機網站二維碼
Copyright ? Yisu Cloud Ltd. All Rights Reserved. 2018 版權所有
廣州億速云計算有限公司粵ICP備17096448號-1 粵公網安備 44010402001142號增值電信業務經營許可證編號:B1-20181529