mmcv 说明及安装步骤
mmcv-full和mmcv-lite
首先要说明的是mmcv-full这个包。在mmcv1.0版本,mmcv分mmcv和mmcvfull,当时的mmcv是cpu版本,而mmcv-full是GPU版本。
但是到2.0版本后,mmcv-lite是cpu版本,而mmcv是GPU版本
mmcv安装的前置条件
mmcv的官网,倒是有说明:安装pytorch和cuda
pytorch和cuda直接使用pytorch官网的命令就直接全装了
另外就是mmcv,似乎是对python版本有要求
一定要看torch版本,我之前没有指定,下载了2.2版本,结果mmcv不支持该版本
查看版本
Installation — mmcv 2.1.0 documentation
mim install mmcv -i https://pypi.tuna.tsinghua.edu.cn/simple
下载指定版本的pytorch,需要指定版本号:
conda install pytorch=2.1.0 torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia