deep-learning - 如何在不支持 cuda 的情况下使 openpose 使用 caffe

Question

我想尝试 Openpose：

https://github.com/CMU-Perceptual-Computing-Lab/openpose

在我的带有 AMD 显卡的笔记本电脑中，所以没有 cuda 是可能的，

这可能吗？如何？

Answer 1

我会在Shai 的回答中补充说您需要禁用

# USE_CUDNN := 1

有时当它离开时，层的设置功能会进行一些 CUDA ASSERT 检查失败并阻止程序继续

Answer 2

使用非 NVIDIA 卡运行 caffe 需要 opencl 分支。将该分支与所使用的 caffe 分支集成OpenPose可能会很棘手（并且可能是直截了当的——我自己没有尝试过）。
如果你想“安全地玩”，你可以通过设置禁用所有 GPU 支持

CPU_ONLY := 1

在你Makefile.config编译 caffe 之前。
这样，您将拥有一个不需要任何 CUDA/NVIDIA 支持的 CPU 版本。

Answer 3

您可以简单地使用最新的 cpu 版本（最新的 openpose 模块，最新的升级）。不需要这样的cuda技术，所以显卡是nvidia还是其他没什么好担心的。

Answer 4

目前没有办法不将 CUDA 与 OpenPose 一起使用。OpenPose 在内部使用 CUDA 代码进行渲染、三次调整大小和非最大抑制。将来，它打算将 CUDA 代码移至 OpenCL（因此它可以在任何显卡上工作），但这是一个长期目标。

已编辑：OpenPose 几个月前添加了仅 CPU 甚至 OpenCL 支持，您可以查看他们的安装文档。