一、安装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 pytorch1.13.1 torchvision0.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 .即可成功