版权声明:本文为博主原创文章,如需转载请附上博文链接 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文件删除来解决。