MhLiao/DB代码链接
cuda版本过高:
程序中要求的环境为pytorch1.2+cuda10.1,但是实验室的显卡为3090ti,3090系列显卡支持的最低cuda版本为11.1,cuda版本过低会报错。
一开始是想完全按照要求的环境来,但是确实没办法降到那么低。后来反应过来,pytorch和cuda不需要一定跟着REDME.md来。
经过了许多试错以后,终于得到可以正常运行的环境:python3.7+pytorch1.8.0+cuda11.3,后来者按照这个环境跑应该是没问题的
附上:cuda多版本安装方法、pytorch官网、cuda官网
pytorch与cuda版本不对应
参考另一篇文章
其他错误
ImportError: /home/ubuntu/DB-master/assets/ops/dcn/deform_conv_cuda.cpython-39-x86_64-linux-gnu.so: undefined symbol: _ZNK2at10TensorBase8data_ptrIdEEPT_v
这是因为修改过torch版本 解决方法:先删除build文件夹,再运行python setup.py build_ext --inplace
Numpy报错:AttributeError: module ‘numpy‘ has no attribute ‘int‘
DB不支持python3.8即以上的numpy,将Numpy换成3.7即可
AttributeError: ‘NoneType‘ object has no attribute ‘astype‘
没获取到图片,查看你图片的链接是否正确