[转载] Pytorch拓展进阶(二)：Pytorch结合C++以及Cuda拓展

最近在看的Pointnet2 pytorch实现版源码中有这样的一段:

看了才知道原来是c++代码

[转载] Pytorch拓展进阶(二)：Pytorch结合C++以及Cuda拓展

看了上面的博客才知道，这些C++有些只是接口，真正的处理写在cu这个CUDA程序文件内(CUDA编程使用的是CUDA C，是C语言的一个超集)

这些文件通过setup.py以下内容配置:

[转载] Pytorch拓展进阶(二)：Pytorch结合C++以及Cuda拓展

原文链接: https://www.cnblogs.com/jiading/p/12187081.html

欢迎关注

微信关注下方公众号，第一时间获取干货硬货；公众号内回复【pdf】免费获取数百本计算机经典书籍

原创文章受到原创版权保护。转载请注明出处：https://www.ccppcoding.com/archives/191901

非原创文章文中已经注明原地址，如有侵权，联系删除

关注公众号【高性能架构探索】，第一时间获取最新文章

转载文章受原作者版权保护。转载请注明原作者出处！