问题描述:
报错情况
分析问题:
- 我既安装了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
3.查询下载相应mmcv版本的命令
点击↑超链接,选择对应torch和cuda进行查询,例如:
然后复制语句到命令行,运行
安装成功,再运行就成功了!