无法在in检测中编译nms_cuda

Question

/usr/local/lib/python3.6/dist-packages/torch/include/ATen/core/TensorBody.h:262:30: note: declared here
   DeprecatedTypeProperties & type() const {
                              ^~~~
mmdet/ops/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 ")
                       ^
mmdet/ops/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 ")
                       ^~~~~~~~
mmdet/ops/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 ")
                       ^
mmdet/ops/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 'x86_64-linux-gnu-gcc' failed with exit status 1

现在我得到的所有信息：

8.2

• /usr/bin/gcc 7
• sys.version
• gcc版本7.5

正常吗？发生了什么？库达错误？gcc出错了吗？如何解决。我被急忙用它来运行‘m检测程序。

Answer 1

解决办法是：

conda install pytorch cudatoolkit==10.0 torchvision -c pytorch -y

如果使用的分支早于v2.0.0。

或者：

conda install pytorch cudatoolkit torchvision -c pytorch -y

如果分支等于v2.0.0。

我花了三天时间才弄明白。基本上，如果您安装了cudatoolkit==10.0，这将迫使pytorch降级为pytorch版本1.4，这使得整个工作正常进行。这一点都不明显。在我看来，这是my检测安装指南中的一个错误。

Answer 2

AT_CHECK不用于新版本的pytorch。

解决方案:尝试将所有AT_CHECK替换为TORCH_CHECK