Linux MMYOLO:环境配置

一、下载源码

二、准备工作

 注意:我们的源码是0.6版本,那么mmdet、mmengine、mmcv必须按照下图要求安装,记得检查

详见依赖 — MMYOLO 0.6.0 文档

mmcv的安装不是那么简单,首先根据源码版本,确定mmcv=2.0.0;然后选择系统、cuda版本、torch版本,得到mmcv的安装命令行。

注意:mmcv版本确定了,那么torch版本不能过高,否则报错,安装torch时格外注意!

详见:Installation — mmcv 2.1.0 documentation

 

接前言,安装torch不要直接在这个表里选,点击进入之前版本,选择正确版本安装:

 

 

三、开始部署 

创建并激活一个 conda 环境:

conda create --name mmyolo python=3.10

conda activate mmyolo

安装pytorch(命令上面有说):

conda install pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pytorch-cuda=11.8 -c pytorch -c nvidia

验证 PyTorch 安装:

安装mim:

pip install -U openmim

根据mmyolo源码下的requirements/mminstall.txt进行安装:

 mim install "mmengine>=0.7.1"

mim install "mmdet>=3.0.0"

mmcv的安装命令行如下(上面有说):
pip install mmcv==2.0.0 -f https://download.openmmlab.com/mmcv/dist/cu118/torch2.0/index.html
mim install -r requirements/albu.txt
mim install -v -e .

四、测试

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值