mmcv安装说明(及pytorch版本对应)

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
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值