MMCV库的正确安装
鉴于MMCV这个库对于新手来说,安装可能比较困难,在这里详细介绍一下MMVC这个库的安装过程
安装过程中,需要保证:cuda与pytorch自己的安装的版本对应
下面是版本对应的链接:参考这个说明书
具体内容包括:首先使用文档中的这个命令查看自己的torch和cuda的版本
python -c 'import torch;print(torch.__version__)'
然后根据自己的torch和cuda的版本安装正确的mmcv,尽量使用下面提供的链接进行安装,一般来说没什么问题.
pip install mmcv==2.1.0 -f https://download.openmmlab.com/mmcv/dist/cu118/torch2.1/index.html
如果还有其他的问题,可以参考这篇文章:MMCV安装指南
Over.