Caffe未使用GPU

问题描述

我已使用以下命令在Python 3.5 anaconda环境中安装了caffe： conda install -c anaconda caffe-gpu。该计算机是已正确配置CUDA的Ubuntu计算机。安装过程没有发生错误，我可以毫无问题地导入caffe。

我正在尝试运行Caffe（https://github.com/BVLC/caffe/tree/master/examples）随附的pascal_multilabel_with_data_layer.ipynb jupyter笔记本示例。我已经下载了PASCAL VOC数据集，并将其放在适当的文件夹（数据文件夹）中。

我已确保在笔记本中将caffe设置为gpu模式，如下所示：

caffe.set_mode_gpu()
caffe.set_device(0)

Nvidia SMI确实显示Caffe在GPU 0上的占用空间符合预期。我仔细阅读了笔记本的所有部分。我确保将solver.prototext文件夹中的models文件设置为将GPU用作solver_mode: GPU。

我通过在虚拟环境中运行以下命令来确保正确配置了caffe： python -c 'import caffe; caffe.set_mode_gpu()' 2>/dev/null && echo "Success" || echo "Failure"。该命令返回成功。

但是，当我运行训练过程时，似乎GPU未使用并且CPU使用率升高，这表明Caffe没有使用GPU进行计算。关于如何进一步调试问题的任何想法？

解决方法

不知道为什么人们在无法回答时会给予不同的帮助。

无论如何，系统python3和conda python3之间存在冲突。康达·福奇的咖啡也不能真正起作用。我必须从源代码重新编译caffe并删除co​​nda RWE限制，才能将软件包直接安装在conda基本目录中。只有这样，它才能正常工作。