安装mmsegmentation框架-有手就行!

安装mmsegmentation框架-有手就行!

框架地址:https://github.com/open-mmlab/mmsegmentation

一.安装pytorch cuda torchvision

1.通过命令行查看驱动版本及可支持的最高CUDA版本

Win+R输入cmd打开命令行,输入命令NVIDIA-smi 查看

2.pytorch 官网 https://pytorch.org/ 复制对应的命令 如下图所示:

在这里插入图片描述


pip3 install install torch==1.10.1+cu113 torchvision==0.11.2+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.htm

我是安装了 PyTorch(1.10.1,针对 CUDA 11.3)和 torchvision(0.11.2,针对 CUDA 11.3)。-f 参数指定了从 PyTorch 官方的特定镜像源下载安装文件。


二.安装openmim mmengine mmcv

pip install -U openmim
mim install mmengine

mmcv与cuda、pytorch版本兼容要求,见mmcv官方文档:
官网:https://mmcv.readthedocs.io/zh-cn/latest/get_started/installation.html#pip
在这里插入图片描述

mim install mmcv==2.0.0

三.安装opencv-python、Pillow、matplotlib、seaborn、tqdm、pytorch-lightning 和 mmdet

pip install opencv-python pillow matplotlib seaborn tqdm pytorch-lightning mmdet>=3.1.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

四.安装 setuptools 和 prettytable

pip install setuptools==59.5 prettytable

setuptools 通常用于Python包的安装和管理,而 prettytable 是一个用于在终端中打印漂亮表格的库。

五.安装 安装 MMSegmentation

git clone https://github.com/open-mmlab/mmsegmentation.git -b v1.2.2#替换你需要的版本 或者自己手动下载
cd mmsegmentation
pip install -v -e .

在这里插入图片描述

跑mask2former试一下
python tools\train.py configs/mask2former/mask2former_swin-b-in1k-384x384-pre_8xb2-160k_ade20k-640x640.py

在这里插入图片描述
安装ftfy

ftfy是一个Python包,‌它的主要功能是自动检测并修复包含错误编码、‌乱码或其他Unicode问题的文本。‌这个工具包可以帮助处理那些由于编码问题导致的文本显示异常,‌通过修复这些问题,‌使得文本能够正确显示。‌

conda install ftfy

看了下预训练权重下载到这个地址了:C:\Users\user.cache\torch\hub\checkpoints
其实可以手动下载 放到pretrain文件夹里
在这里插入图片描述

在这里插入图片描述
成功!!!!!

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值