mmfewshot 安装踩坑记录

最新推荐文章于 2024-07-04 09:37:15 发布

weixin_39260994

最新推荐文章于 2024-07-04 09:37:15 发布

阅读量1.3k

点赞数 1

文章标签： python 深度学习 pytorch

本文链接：https://blog.youkuaiyun.com/weixin_39260994/article/details/126324782

版权

mmfewshot 安装踩坑记录

安装流程

安装流程

cuda 环境：10.1
在这里插入图片描述
https://github.com/open-mmlab/mmfewshot/blob/main/docs/zh_cn/install.md

conda环境搭建

conda create -n openmmlab python=3.7 -y
conda activate openmmlab

conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=10.1 -c pytorch

安装MMFewShot

安装 mmcv-full

pip install mmcv-full==1.6.0 -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.7.0/index.html

$\color{red}{注意mmcv-full一定得下载1.6.0以下的版本，否则会报错：}$
在这里插入图片描述

安装MMClassification 和 MMDetection

pip install mmcls
pip install mmdet==2.25.0 -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.7.0/index.html

$\color{red}{注意mmdet一定得下载2.25.0以下的版本，否则会报错：}$
在这里插入图片描述

安装 MMFewShot

pip install mmfewshot -i https://mirrors.tuna.tsinghua.edu.cn/help/centos/
git clone https://github.com/open-mmlab/mmfewshot.git
cd mmfewshot

pip install -r requirements/build.txt
pip install -v -e .  # or "python setup.py develop"