【解决】报错：部署Deformable-DETR环境问题
阅读readme发现需要在 model/opts/下 sh make.sh 安装环境。最初始时报错：没有 MSDeformAttnFunction 这个module。又报了一大长段错，找到一个错误行。在配置文件中添加如下一行。保存后source更新。再运行程序，问题解决。
·
最初始时报错：没有 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
再运行程序，问题解决
有“AI”的1024 = 2048，欢迎大家加入2048 AI社区
更多推荐
0
0- 0
已为社区贡献1条内容
所有评论(0)