Fatal编程技术网
  • python/
  • Caffe&x27；s PythonLayer支持GPU吗？

Caffe&x27；s PythonLayer支持GPU吗？

python

Caffe&x27；s PythonLayer支持GPU吗？,python,gpu,caffe,Python,Gpu,Caffe,CaffePythonLayer的前向和/或后向通道是否可以访问GPU 最近有人在这里问这个问题： 更新 这是可能的合并的 在PR 5904之前 从和引用的拉取请求， 似乎（在撰写本文时），Caffe不支持在自定义Python层之间传递GPU数据。由于PythonLayer是完整的Python，因此它可以访问GPU。它只需要在CPU和GPU之间多次复制数据。仍然可以从Python层使用GPU。请参阅通过层外部配置键执行此操作的rcnn repo：

Caffe
PythonLayer
的前向和/或后向通道是否可以访问GPU

最近有人在这里问这个问题：

更新 这是可能的合并的

在PR 5904之前 从和引用的拉取请求，
似乎（在撰写本文时），Caffe不支持在自定义Python层之间传递GPU数据。由于
PythonLayer
是完整的Python，因此它可以访问GPU。它只需要在CPU和GPU之间多次复制数据。

仍然可以从Python层使用GPU。请参阅通过层外部配置键执行此操作的rcnn repo：