Caffe 不使用 GPU

Question

我已经使用以下命令在 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 notebook 示例。我已经下载了 PASCAL VOC 数据集并放入相应的文件夹（数据文件夹）。

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

caffe.set_mode_gpu()
caffe.set_device(0)

Nvidia SMI 确实显示 Caffe 按预期占用 GPU 0 上的空间。我浏览了笔记本的所有部分，没有错误。我确保models 文件夹中的solver.prototext 文件设置为使用GPU 作为solver_mode: GPU。

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

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

Answer 1

不知道为什么人们在无法回答时会给出不同的意见。

无论如何，system python3和conda python3之间存在冲突。 conda forge 的 caffe 也不起作用。我必须从源代码重新编译 caffe 并删除 conda RWE 限制才能将软件包直接安装在 conda 基目录中。只有这样它才能正常工作。