RP-R-CNN
RP-R-CNN copied to clipboard
Published
20 hours ago •
soeaver
soeaver
Error installing RP-R-CNN using google colab
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
