machine-learning - 是否可以使用具有编译设置 USE_CUDNN := 1? 的 CPP 代码

Question

在caffe中，编译时我必须选择设置USE_CUDNN := 1。我将使用 .hpp 和 .cpp 代码(不含 .cu)编写一个新层。如果我没有编写 cuda 代码 (.cu)，我仍然可以使用我的图层吗？例如，我将使用 engine=CAFFE

设置图层

Answer 1

您可能有仅具有 CPU 实现的层。请参阅development页面，了解需要进行哪些更改才能使您的层成为“仅限 CPU”。

但是，请注意，当通过网络进行转发/反向传播时，每次数据通过层时，数据都会从 GPU 复制到 CPU 内存。这种来回的内存同步可能会花费大量时间，并严重阻碍您的计算。在大多数情况下，值得付出努力并为该层实现 GPU 版本。