Caffe的PythonLayer是否支持GPU？

Question

Caffe PythonLayer的前进和/或后退传递是否可以访问GPU？

这个问题最近在这里问： https://github.com/BVLC/caffe/issues/5286

Answer 1

更新

这可以通过合并Pull Request 5904。

来实现

在PR 5904之前

来自Caffe-Users mailing和引用的拉取请求2102， 看起来（在撰写本文时），Caffe不支持在自定义Python层之间传递GPU数据。由于PythonLayer是完整的Python，它可以访问GPU。它只需要在CPU和GPU之间多次复制数据。