如何使用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内存使用情况”不足以适合我的模型（因此我知道错误...）

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。