在使用如下命令安裝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正常運行