在使用如下命令安装TubeTK的cuda_nms时,报了一个错误,记录一下这个错误和解决办法
(base) redmery@redmery:~/Desktop/MOT/TubeTK/post_processing/nms$ python setup.py build_ext --inplace
因为这个命令是在/home/redmery/Desktop/MOT/TubeTK/install/compile_local.sh中的
直接将编译的代码在命令行中编译
报错:
从报错中查看,发现里面有at_check,因此搜索后,在error: ‘AT_CHECK’ was not declared in this scope解决方法中发现了解决办法,将编译文件nms_cuda.cpp中的at_check改为torch_check即
改为
编译通过,TubeTK就可以正常使用了
TubeTK正常运行