如何使用GPU确保caffe？

Question

有没有办法确保使用GPU的caffe？在cmake中安装CUDA驱动程序并且没有CPU_ONLY标志并且在编译cmake记录的CUDA 8.0检测时，我编译了caffe。

但是在训练样本时，我怀疑它是根据nvidia-smi结果使用GPU。我怎样才能确保？

Answer 1

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

solver_mode：GPU

如果您在模型的每个层中都有任何引擎规格，那么您还需要确保它们引用GPU软件。

Answer 2

对于未来的咖啡流浪者来说，这终于为我做了诀窍：

caffe.set_mode_gpu()
caffe.set_device(0)

我确实有solver_mode: GPU，它会在gpu上显示进程，但使用nvidia-smi看到的'GPU Memory Usage'不足以适合我的模型（所以我知道出了什么问题......）

Answer 3

您可以在程序中明确使用Caffe::set_mode(Caffe::GPU);。

要确保该进程使用GPU，您可以在ubuntu中使用nvidia-smi命令将进程使用GPU。

至于我，我使用MTCNN进行人脸检测（由caffe实现）：

我使用nvidia-smi命令来显示使用GPU的进程，如果你想通过间隔使用watch nvidia-smi来查看它。

如下图所示，我们可以看到进程mtcnn_c(use caffe backend)正在使用GPU。