[转载] Pytorch拓展进阶(二)：Pytorch结合C++以及Cuda拓展
来源：https://oldpan.me/archives/pytorch-cuda-c-plus-plus
最近在看的Pointnet2 pytorch实现版源码中有这样的一段:
看了才知道原来是c++代码
看了上面的博客才知道，这些C++有些只是接口，真正的处理写在cu这个CUDA程序文件内(CUDA编程使用的是CUDA C，是C语言的一个超集)
这些文件通过setup.py以下内容配置:
[转载] Pytorch拓展进阶(二)：Pytorch结合C++以及Cuda拓展
猜你喜欢
转载自www.cnblogs.com/jiading/p/12187081.html
今日推荐
周排行