WIN10 下配置 mmdetection 2.18
硬件环境 3090 + i9-10900k 软件环境 cuda 11.4 + VS2019问题描述:
mmdetection在win10下的安装可能存在着各种各样的bug,可能随着版本更新会存在一些变化,需要根据实际情况修改
目前针对最新的mmdetection 2.18 版本在win10 下的安装总结一下流程,帮助大家更好的在 win10 下安装 mmdetection
https://mmdetection.readthedocs.io/zh_CN/latest/get_started.html#id2 #这个是官方的安装教程,大家可以参考
https://github.com/open-mmlab/mmdetection/blob/master/docs/useful_tools.md #训练中的有用工具
按照官方的依赖来,检查软件环境
安装流程:
大家参照官方的教程进行安装,下面是重点要说明的地方
使用 conda 新建虚拟环境,并进入该虚拟环境
conda create -n open-mmlab python=3.7 -y
conda activate open-mmlab
!!!重要的事情,python版本要3.7的,原来尝试装了3.9的版本报了莫名奇妙的错误(还找不出来,以为是官方不支持window导致的)
下面在 anaconda 中安装 pytorch,在https://pytorch.org/上可以查看 pytorch 版本和 cuda 的对应关系,并生成安装命令
没有看到 cuda11.4 的选项就选了 11.3,cudatoolkit 的版本低于电脑安装的版本(我电脑安装的是11.4)应该是没有影响的
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
!!!安装时有时候会下载的很慢,有一个有效的方法就是开一个手机热点去下载(大概2G流量),我这边用流量下载要快很多(10KB到1MB的差距)
安装 MMDetection的步骤时,我采用的是下面这个方法 (教程里的第一种方法我这边尝试了会报错)
(与上面安装的对应我这里就是 cu_version=cu113,torch_version=1.10.0)
pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/{cu_version}/{torch_version}/index.html
pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu113/torch1.10.0/index.html #我的安装指令
安装mmdet,我就是直接pip安装最新版本
pip install mmdet
要安装额外的依赖以使用 Instaboost, 全景分割, 或者 LVIS 数据集 参照官方安装教程(我暂时用不到就没有安装)
# 安装 instaboost 依赖
pip install instaboostfast
# 安装全景分割依赖
pip install git+https://github.com/cocodataset/panopticapi.git
# 安装 LVIS 数据集依赖
pip install git+https://github.com/lvis-dataset/lvis-api.git
# 安装 albumentations 依赖
pip install albumentations>=0.3.2 --no-binary imgaug,albumentations
这样就完成了mmdetection 2.18的安装了