问题描述:
报错情况
分析问题:
- 我既安装了mmcv也安装了mmcv-full,其实mmcv是mmcv-full的简装版。所以只需要安装一个即可。
- 其次,我卸载了mmcv-full,仍然报错,考虑是mmcv版本的问题。
- 但是,通过nvidia-smi查询的cuda版本安装的mmcv不正确,考虑cuda版本不正确。
解决方法:
1.卸载已安装的mmcv和mmcv-full
pip uninstall mmcv
pip uninstall mmcv-full
2.查看cuda和torch的版本
python -c 'import torch;print(torch.__version__);print(torch.version.cuda)'
结果如图:
由此可见torch=1.12.1,cuda=10.2

文章讲述了用户在安装mmcv时遇到的错误,推测是由于mmcv版本与CUDA版本不匹配。作者提供了卸载旧版本、检查cuda和torch版本以及找到对应mmcv版本的解决方案,最终成功安装并解决问题。

最低0.47元/天 解锁文章
66

被折叠的 条评论
为什么被折叠?



