libtorch error C1021: 无效的预处理器命令“warning”

 

编译：py_utils的_cpools时出现问题，

 

python setup.py build develop

 

 

D:\ProgramData\Miniconda3\lib\site-packages\torch\include\torch\csrc\api\include\torch/torch.h(7): fatal error C1021: 无效的预处理器命令“warning”
error: command 'C:\\Program Files (x86)\\Microsoft Visual Studio\\2017\\Professional\\VC\\Tools\\MSVC\\14.16.27023\\bin\\HostX86\\x64\\cl.exe' failed with exit status 2

感谢下面的牛人：下面这个群是作者的，不是我的：

WIN10下已经解决了安装和训练，还有问题的，可以加Q群264191384 注明（CornerNet-Lite）共同研究

 

解决方法：

#include <torch/torch.h>

改为：#include <torch/extension.h>