In file included from /root/anaconda3/envs/py39/lib/python3.9/site-packages/torch/include/torch/csrc/Device.h:4:0,
from /root/anaconda3/envs/py39/lib/python3.9/site-packages/torch/include/torch/csrc/api/include/torch/python.h:8,
from /root/anaconda3/envs/py39/lib/python3.9/site-packages/torch/include/torch/extension.h:9,
from /home/ma-user/work/jinzili/code/detectron2_repo/detectron2/layers/csrc/vision.cpp:3:
/root/anaconda3/envs/py39/lib/python3.9/site-packages/torch/include/torch/csrc/python_headers.h:12:10: fatal error: Python.h: No such file or directory
#include <Python.h>
^~~~~~~~~~
compilation terminated.
...
File "/root/anaconda3/envs/py39/lib/python3.9/site-packages/torch/utils/cpp_extension.py", line 1774, in _write_ninja_file_and_compile_objects
_run_ninja_build(
File "/root/anaconda3/envs/py39/lib/python3.9/site-packages/torch/utils/cpp_extension.py", line 2116, in _run_ninja_build
raise RuntimeError(message) from e
RuntimeError: Error compiling objects for extension
[end of output]
note: This error originates from a subprocess, and is likely not a problem with pip.
解决办法:
export CPATH="$CONDA_PREFIX/include/python3.9:$CPATH"
export CPLUS_INCLUDE_PATH="$CONDA_PREFIX/include/python3.9:$CPLUS_INCLUDE_PATH"