command gcc failed with exit status 1报错

command ‘gcc’ failed with exit status 1`报错

报错!头疼!

econn/models/operators/nms/src/nms_cuda.cpp:4:23: error: ‘AT_CHECK’ was not declared in this scope
 #define CHECK_CUDA(x) AT_CHECK(x.type().is_cuda(), #x, " must be a CUDAtensor ")
                       ^
econn/models/operators/nms/src/nms_cuda.cpp:4:23: note: in definition of macro ‘CHECK_CUDA’
 #define CHECK_CUDA(x) AT_CHECK(x.type().is_cuda(), #x, " must be a CUDAtensor ")
                       ^~~~~~~~
econn/models/operators/nms/src/nms_cuda.cpp:4:23: note: suggested alternative: ‘DCHECK’
 #define CHECK_CUDA(x) AT_CHECK(x.type().is_cuda(), #x, " must be a CUDAtensor ")
                       ^
econn/models/operators/nms/src/nms_cuda.cpp:4:23: note: in definition of macro ‘CHECK_CUDA’
 #define CHECK_CUDA(x) AT_CHECK(x.type().is_cuda(), #x, " must be a CUDAtensor ")
                       ^~~~~~~~
error: command 'gcc' failed with exit status 1

尝试:

1. sudo apt-get install python3.7-dev 网上找的
2. 将torch版本从1.5降到1.2 不行
3. 将cuda版本从10.0跟新到10.1

cuda安装地址:
https://developer.nvidia.com/cuda-10.1-download-archive-base