是否可以使用CPP代码和编译设置USE_CUDNN：= 1？

Question

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

设置图层

Answer 1

您可能拥有仅具有CPU实施的图层。有关使您的图层成为“仅限CPU”所需更改的说明，请参阅development页面。

但请注意，在通过网络转发/反向传输时，每次数据通过您的图层时，数据都会从GPU复制到CPU内存。这种来回内存同步可能会花费大量时间并显着停止计算。在大多数情况下，同样努力并为该层实现GPU版本也是值得的。