2016-11-10 84 views
0

有沒有什麼辦法可以確保使用GPU的caffe?我在安裝CUDA驅動程序後編譯了caffe,並且沒有在cmake中標記CPU_ONLY,並編譯cmake記錄了CUDA 8.0的檢測。如何確保使用GPU的咖啡?

但是訓練樣本時,我懷疑它使用GPU根據nvidia-smi的結果。我如何確保?

+2

如果你沒有看到caffe在'nvidia-smi'中運行,它可能不會使用GPU。你有沒有在你的solver.prototxt中設置solver_mode:gpu? – Shai

回答

1

我知道最可靠的方法是正確配置solver.prototxt文件。 包括行

solver_mode:GPU

如果您有任何規格的引擎在模型的每一層使用,你也想確保他們指的GPU軟件。