Caffe添加自定义的层遇到的一个问题及解决方案

版权声明:本文为博主原创文章,如需转载请附上博文链接 https://blog.csdn.net/wendygelin/article/details/88427560

在学习自定义层的过程中,发现一个可能的坑

自定义层:

https://blog.csdn.net/wfei101/article/details/76735760

https://blog.csdn.net/u012426298/article/details/81016600

我在自定义的层中使用GPU果然出现错误,先不管,只用CPU,发现编译仍然不通过。

该问题通过,将之前同时导入的.cu文件删除来解决。

猜你喜欢

转载自blog.csdn.net/wendygelin/article/details/88427560
今日推荐