ERROR: torch/extension.h and ATen/ATen.h

[whdong-whu]

when i run make.sh,i get this error
'''
deformable_conv2d.cc:1:10: fatal error: torch/extension.h: No such file or directory
1 | #include <torch/extension.h>
| ^~~~~~~~~~~~~~~~~~~
compilation terminated.
deformable_conv2d_cuda.cu:3:10: fatal error: ATen/ATen.h: No such file or directory
3 | #include <ATen/ATen.h>
'''
my environment list:
python 3.6
pytorch 1.7.1
cuda 11.2
cuda_tookit 11.1
thanks a lot