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
Logo

更多推荐