graph-rcnn.pytorch
graph-rcnn.pytorch copied to clipboard
Compilation stage failure
When compiling lib/scene_parser/rcnn got error: ... lib/python3.7/site-packages/torch/include/ATen/Functions.h:4128:229: required from here /usr/include/c++/6/tuple:495:244: error: wrong number of template arguments (4, should be 2) return _and<_not<is_same<tuple<_Elements...>, ... error: command '/usr/lib/nvidia-cuda-toolkit/bin/nvcc' failed with exit status 1
Modify setup.py (add -ccbin=/usr/bin/gcc-5):
extra_compile_args["nvcc"] = [ "-DCUDA_HAS_FP16=1", "-D__CUDA_NO_HALF_OPERATORS__", "-D__CUDA_NO_HALF_CONVERSIONS__", "-D__CUDA_NO_HALF2_OPERATORS__", "-ccbin=/usr/bin/gcc-5", ]