DBNet工程:https://github.com/MhLiao/DB
执行echo $CUDA_HOME cd assets/ops/dcn/ python setup.py build_ext --inplace
编译deformable convolution出错:
-
nvcc fatal : Unsupported gpu architecture ‘compute_86’
解决方案:cuda版本支持不了这么高的算力(8.6),降低算力要求export TORCH_CUDA_ARCH_LIST="7.5"
参考:https://blog.csdn.net/qq_30614451/article/details/111173703
-
‘AT_CHECK’ was not declared in this scope
解决方案:AT_CHECK 在torch 1.5中弃用了,后续版本用TORCH_CHECK替用。
所以将代码中所有的的‘AT_CHECK’全部替换为‘TORCH_CHECK’即可编译成功。
参考:https://blog.csdn.net/qq_21388689/article/details/117129404