win10安装配置mmdetection教程

一、安装Anaconda并创建虚拟环境
1、官网下载Anaconda安装包安装即可
2、创建新的虚拟环境
conda create -n openmmlab python=3.8
conda activate openmmlab

二、配置Pytorch环境
1、查看并确定自己电脑的CUDA版本
打开nvidia设置面板,点击【帮助】-【系统信息】

选择【组件】在【3D设置】里面可以看到cuda版本,这里是11.7

2、去Pytorch官网https://pytorch.org/get-started/previous-versions/找对应cuda版本的安装命令

我的这里用的是:
conda install pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 pytorch-cuda=11.7 -c pytorch -c nvidia
3、安装pycocotools-windows
pip install pycocotools-windows
4、按照mmdetection文档中的步骤进行后续安装
https://mmdetection.readthedocs.io/zh_CN/latest/get_started.html
1)安装mim、mmengine、mmcv
pip install -U openmim
mim install mmengine
mim install "mmcv>=2.0.0"
2)从源码安装mmdetection
git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
pip install -v -e .
【最后一步会报错,无法build pycocotools,解决方法在下面第五步中讲】
5、去VS官网https://visualstudio.microsoft.com/zh-hans/visual-cpp-build-tools/下载安装生成工具
1)下载生成工具

下载好后运行,选择左上角【使用C++的桌面开发】

右侧选项默认就好(如果默认里面没有win10 sdk的话记得选上)
然后点击开始安装,等待结束
安装完成后点击【启动】

然后重新运行【第四步】中的pip install -v -e .即可成功

posted @ 2023-05-27 12:58  彼岸的客人  阅读(378)  评论(0编辑  收藏  举报