mmcv 报错undefined symbol: _ZNK2at6Tensor7is_cudaEv

最新推荐文章于 2025-04-07 22:20:05 发布

幸福回头

最新推荐文章于 2025-04-07 22:20:05 发布

阅读量9.7k

点赞数 1

分类专栏：系统相关图像处理文章标签：人工智能

本文链接：https://blog.youkuaiyun.com/zt1091574181/article/details/121773706

版权

图像处理同时被 2 个专栏收录

13 篇文章

订阅专栏

系统相关

12 篇文章

订阅专栏

这篇博客讨论了在Python环境中遇到的mmcv库导入错误，错误信息提示缺失符号。问题根源在于torch和CUDA版本不匹配。解决方法是参照官方链接提供的兼容性表，选择适合的cuda和torch版本进行安装，确保两者协调一致。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

>>> from mmcv.ops import nms
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "/root/.local/lib/python3.7/site-packages/mmcv/ops/__init__.py", line 2, in <module>
from .assign_score_withk import assign_score_withk
File "/root/.local/lib/python3.7/site-packages/mmcv/ops/assign_score_withk.py", line 6, in <module>
'_ext', ['assign_score_withk_forward', 'assign_score_withk_backward'])
File "/root/.local/lib/python3.7/site-packages/mmcv/utils/ext_loader.py", line 13, in load_ext
ext = importlib.import_module('mmcv.' + name)
File "/root/.local/conda/envs/open-mmlab/lib/python3.7/importlib/__init__.py", line 127, in import_module
return _bootstrap._gcd_import(name[level:], package, level)
ImportError: /root/.local/lib/python3.7/site-packages/mmcv/_ext.cpython-37m-x86_64-linux-gnu.so: undefined symbol: _ZNK2at6Tensor7is_cudaEv

出现这样的错误主要是由于安装mmcv的时候需要对torch版本和cuda版本同时进行限制，

https://download.openmmlab.com/mmcv/dist/cu101/torch1.5.0/index.html

如上面的链接：

需要设置好cuda版本、torch版本进行下载