可能是编译过程中使用的torch版本与cuda版本不对应的问题,可以使用torch.version.cuda查看torch对应的CUDA版本。
解决方案:https://github.com/sfzhang15/ATSS/issues/17