参考网址:安装 MMCV — mmcv 2.1.0 文档https://mmcv.readthedocs.io/zh-cn/latest/get_started/installation.html
1、 安装 mmcv 前,确保 PyTorch 已经成功安装在环境中,可使用以下命令进行验证:
python -c 'import torch;print(torch.__version__)'
2、 使用 min 安装
pip install -U openmim
mim install mmcv
如果安装时间过长,可以制定 pypi 源
mim install "mmcv>=2.0.0rc1" -i https://pypi.tuna.tsinghua.edu.cn/simple
3、使用 pip 安装
查看 CUDA 和 PyTorch 的版本
python -c 'import torch;print(torch.__version__);print(torch.version.cuda)'
pip install mmcv==2.1.0 -f https://download.openmmlab.com/mmcv/dist/cu117/torch1.13/index.html
安装成功咯!