我可以訓練一個沒有GPU的深度卷積網絡嗎？

Question

我想構建一個卷積神經網絡作爲跟蹤系統應用程序。我感覺所有深層網絡應用程序都需要使用GPU。是否有必要在像我這樣的任務中使用GPU？我的筆記本電腦的最低PC要求是什麼？

Answer 1

這一切都取決於你的CNN的大小和深度。如果您的CNN有一個卷積圖層和一個完全連接的圖層，並且輸入圖像爲64x64，則您可以在合理的時間內在筆記本電腦上訓練您的網絡。如果您使用GoogLeNet和上百個圖層，並且在整個ImageNet集上進行訓練，即使使用視頻卡也需要一週的時間，所以在CPU上它永遠不會完成訓練。

然而，對於大多數實際應用，希望有一個GPU來訓練卷積網絡。請注意，在AWS上，您可以以相當合理的價格獲得支持GPU的實例，特別是在獲得專有實例的情況下，因此您不一定需要在本地擁有GPU。最後說明：大多數框架（theano，torch，caffe，mxnet，tensorflow）允許您在CPU和GPU上執行相同的模型，只需稍微修改或不修改代碼，因此您可以在具有一小組圖像的CPU，然後在您的模型運行時，在GPU上對AWS實例進行訓練。