Error installing RP-R-CNN using google colab

[mrtnzrm2]

Hello!

I am trying to install RP-R-CNN in google colab, but I am having a problem with the sh make.sh step. When running I am getting the following message:

/usr/local/cuda/bin/nvcc -DWITH_CUDA -I/content/RP-R-CNN/models/ops/csrc -I/usr/local/lib/python3.6/dist-packages/torch/include -I/usr/local/lib/python3.6/dist-packages/torch/include/torch/csrc/api/include -I/usr/local/lib/python3.6/dist-packages/torch/include/TH -I/usr/local/lib/python3.6/dist-packages/torch/include/THC -I/usr/local/cuda/include -I/usr/include/python3.6m -c /content/RP-R-CNN/models/ops/csrc/cuda/deform_pool_cuda.cu -o build/temp.linux-x86_64-3.6/content/RP-R-CNN/models/ops/csrc/cuda/deform_pool_cuda.o -D__CUDA_NO_HALF_OPERATORS__ -D__CUDA_NO_HALF_CONVERSIONS__ -D__CUDA_NO_HALF2_OPERATORS__ --expt-relaxed-constexpr --compiler-options '-fPIC' -DCUDA_HAS_FP16=1 -D__CUDA_NO_HALF_OPERATORS__ -D__CUDA_NO_HALF_CONVERSIONS__ -D__CUDA_NO_HALF2_OPERATORS__ -DTORCH_API_INCLUDE_EXTENSION_H -DTORCH_EXTENSION_NAME=_C -D_GLIBCXX_USE_CXX11_ABI=0 -gencode=arch=compute_75,code=sm_75 -std=c++14
/content/RP-R-CNN/models/ops/csrc/cuda/deform_pool_cuda.cu(42): error: identifier "AT_CHECK" is undefined

/content/RP-R-CNN/models/ops/csrc/cuda/deform_pool_cuda.cu(68): error: identifier "AT_CHECK" is undefined

2 errors detected in the compilation of "/tmp/tmpxft_00000a06_00000000-6_deform_pool_cuda.cpp1.ii".
error: command '/usr/local/cuda/bin/nvcc' failed with exit status 1

It appears that there is a problem with deform_pool_cuda.cu.

In another git issue, I found that adding

#ifndef AT_CHECK
#define AT_CHECK TORCH_CHECK 
#endif

at the top of deform_pool_cuda.cu would solve the problem, but it is still happening to me.

Could you check it out? or give me some references or hints, please?

The computer has:

Python 3.6.9

nvcc: NVIDIA (R) Cuda compiler driver
Cuda compilation tools, release 10.1, V10.1.243

Pytorch: 1.7.0+cu101

gcc (Ubuntu 5.5.0-12ubuntu1) 5.5.0 20171010

[abhigoku10]

@mrtnzrm2 any progress on this thread i am also facign the same issue

[jiawenhao2015]

pytorch version should be 1.4.0