由于这个bug调的时间有点长,很多参考的链接追溯不了,总的来说比较重要的链接是:
https://github.com/jwyang/faster-rcnn.pytorch/issues/503
我的过程就凭记忆简单总结一下:
1. 我这里显示找不到‘_C’文件
cuda和pytorch版本的问题,具体是哪个忘了,好像把cuda took..那个下在conda环境里不太行。于是我整了多版本CUDA,成功的是CUDA10.0+pytorch1.0,其中遇到的幺蛾子整理了另一篇,见:Ubuntu22.04多版本CUDA切换,并安装CUDA10.0-CSDN博客
2.报错2:
于是我bug换了一个。。。
File "xxxxxxxxxx/STTran-main/fasterRCNN/lib/model/roi_layers/nms.py", line 3, in <module>
from fasterRCNN.lib.model import _C
ImportError: /xxxxxxxxxx/STTran-main/fasterRCNN/lib/model/_C.cpython-36m-x86_64-linux-gnu.so: undefined symbol: _ZN2at19UndefinedTensorImpl10_singletonE
意思是文件找到了,没想要的
其实是我复现的代码,人家生成的‘_C’文件我直接拿来用是不行的,得用自己的环境生成
rm -rf lib/build/
python setup.py build develop
报错消除~