一、下载与环境配置
1.1下载mmyolo
mmyolo下载路径:https://github.com/open-mmlab/mmyolo
1.2环境配置
主要参照mmyolo文档https://mmyolo.readthedocs.io/en/latest/get_started/dependencies.html
https://github.com/open-mmlab/mmyolo?tab=readme-ov-file(我的建议是参考这个)
首先,在官方文档中查看环境要求;
conda create -n mmyolo python=3.8 pytorch==1.10.1 torchvision==0.11.2 cudatoolkit=11.3 -c pytorch -y
conda activate mmyolo
#mmlab工具
pip install openmim
mim install "mmengine>=0.6.0"
mim install "mmcv>=2.0.0rc4,<2.1.0"
mim install "mmdet>=3.0.0,<4.0.0"
#克隆mmyolo
git clone https://github.com/open-mmlab/mmyolo.git
#进入mmyolo文件
cd mmyolo
# Install albumentations
pip install -r requirements/albu.txt
# Install MMYOLO
mim install -v -e .
但要求中的环境配置需要做出调整,不能直接调用这个语句,以下是虚拟环境配置的整体流程;
#创建python=3.8的虚拟环境,名称为mmyolo
conda create -n mmyolo python=3.8
#激活环境
conda activate mmyolo OR activate mmyolo
#查看库中cudatoolkit版本
conda search cudatoolkit
#下载cudatoolkit
conda install cudatoolkit==11.3.1
#查看库中cudnn版本
conda search cudnn
#下载cudnn
conda install cudnn==8.2.1
#再去拉官方文档中代码,下载torch等
conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch
如果你下载的时候很慢,甚至报错,很可能是你的channels有问题,所以我们要做Anaconda换源;
#清除配置的源,恢复默认状态
conda config --remove-key channels
#添加源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config -