编译Pytorch Custom C++ and CUDA Extension出错问题

项目场景：

做项目时用到了ICP算法，需要用到CUDA加速，但是编译Pytorch C++拓展时出错了。

---

问题描述

使用

python setup.py build develop

时出现

g++: error: vision.o: No such file or directory
error: command '/usr/bin/g++' failed with exit code 1

---

原因分析：

对Pytorch C++ Extension进行编译时会使用到Ninja加快编译，但是目前还不知道是什么原因使用python setup.py build develop ninja不会自动运行，于是就没有.o文件。

---

解决方案：

手动运行ninja编译C++文件生成.o文件

ninja -f YourPath/build.nin