Caffe 不使用 GPU
Caffe not using GPU
我已经使用以下命令在 Python 3.5 anaconda 环境中安装了 caffe：
conda install -c anaconda caffe-gpu。该机器是一台已经正确配置了 CUDA 的 Ubuntu 机器。安装顺利进行，我可以毫无问题地导入 caffe。
我正在尝试 运行 pascal_multilabel_with_data_layer.ipynb Caffe (https://github.com/BVLC/caffe/tree/master/examples) 提供的 jupyter notebook 示例。我已经下载了PASCAL VOC数据集，并放在了合适的文件夹（数据文件夹）中。
我已经确保笔记本中的caffe设置为gpu模式如下：
caffe.set_mode_gpu()
caffe.set_device(0)
Nvidia SMI 确实按预期显示 Caffe 在 GPU 0 上占用 space。我毫无错误地浏览了笔记本的所有部分。我确保 models 文件夹中的 solver.prototext 文件设置为使用 GPU 作为 solver_mode: GPU。
我通过 运行 在虚拟环境中执行以下命令确保 caffe 已正确配置：
python -c 'import caffe; caffe.set_mode_gpu()' 2>/dev/null && echo "Success" || echo "Failure"。命令returns成功。
然而，当我 运行 训练过程时，似乎 GPU 未被使用并且 CPU 使用率很高，这表明 Caffe 没有使用 GPU 进行计算。关于如何进一步调试问题的任何想法？
不知道为什么人们在无法回答时会点赞。
无论如何，系统 python3 和 conda python3 之间发生了冲突。 conda forge 的 caffe 也不起作用。我不得不从源代码重新编译 caffe 并删除 conda RWE 限制以直接在 conda 基本目录中安装包。只有这样它才能正常工作。
我已经使用以下命令在 Python 3.5 anaconda 环境中安装了 caffe：
conda install -c anaconda caffe-gpu。该机器是一台已经正确配置了 CUDA 的 Ubuntu 机器。安装顺利进行，我可以毫无问题地导入 caffe。
我正在尝试 运行 pascal_multilabel_with_data_layer.ipynb Caffe (https://github.com/BVLC/caffe/tree/master/examples) 提供的 jupyter notebook 示例。我已经下载了PASCAL VOC数据集，并放在了合适的文件夹（数据文件夹）中。
我已经确保笔记本中的caffe设置为gpu模式如下：
caffe.set_mode_gpu()
caffe.set_device(0)
Nvidia SMI 确实按预期显示 Caffe 在 GPU 0 上占用 space。我毫无错误地浏览了笔记本的所有部分。我确保 models 文件夹中的 solver.prototext 文件设置为使用 GPU 作为 solver_mode: GPU。
我通过 运行 在虚拟环境中执行以下命令确保 caffe 已正确配置：
python -c 'import caffe; caffe.set_mode_gpu()' 2>/dev/null && echo "Success" || echo "Failure"。命令returns成功。
然而，当我 运行 训练过程时，似乎 GPU 未被使用并且 CPU 使用率很高，这表明 Caffe 没有使用 GPU 进行计算。关于如何进一步调试问题的任何想法？
不知道为什么人们在无法回答时会点赞。
无论如何，系统 python3 和 conda python3 之间发生了冲突。 conda forge 的 caffe 也不起作用。我不得不从源代码重新编译 caffe 并删除 conda RWE 限制以直接在 conda 基本目录中安装包。只有这样它才能正常工作。