我已使用以下命令在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进行计算。关于如何进一步调试问题的任何想法?
答案 0 :(得分:0)
不知道为什么人们在无法回答时会给予不同的帮助。
无论如何,系统python3和conda python3之间存在冲突。康达·福奇的咖啡也不能真正起作用。我必须从源代码重新编译caffe并删除conda RWE限制,才能将软件包直接安装在conda基本目录中。只有这样,它才能正常工作。