【解决】报错：部署Deformable-DETR环境问题_no module named 'deformable_detr_ops

作者：花生_TL007 | 2024-04-16 13:12:12

踩

no module named 'deformable_detr_ops

最初始时报错：没有 MSDeformAttnFunction 这个module
阅读readme发现需要在 model/opts/下 sh make.sh 安装环境
sh 之后又报错
试了一下在别人评论区看到的方法：
pip install ./ops
又报了一大长段错，找到一个错误行
nvcc fatal : Unsupported gpu architecture 'compute_86'
遂百度：https://blog.csdn.net/qq_31347869/article/details/123348901

打开配置文件
vi ~/.bashrc
在配置文件中添加如下一行
export TORCH_CUDA_ARCH_LIST="8.0" # 因为是CUDA11.0，对应的算力为8.0
保存后source更新
source ~/.bashrc

再运行程序，问题解决

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/花生_TL007/article/detail/434379