mmdetection环境安装
时间: 2024-05-15 19:08:52 浏览: 8
以下是mmdetection环境安装的步骤:
1.安装Anaconda:可以从官网下载对应版本的Anaconda进行安装。
2.使用Conda创建虚拟环境:在Anaconda Prompt中输入以下命令创建名为mmdet的虚拟环境。
```
conda create -n mmdet python=3.7 -y
```
3.激活虚拟环境:在Anaconda Prompt中输入以下命令激活mmdet虚拟环境。
```
conda activate mmdet
```
4.安装PyTorch:可以在PyTorch官网上选择对应的版本进行安装,也可以使用以下命令安装。
```
conda install pytorch==1.9.0 torchvision==0.10.0 torchaudio==0.9.0 cudatoolkit=11.1 -c pytorch -y
```
5.安装MMCV:可以使用以下命令安装。
```
pip install mmcv-full==1.5.1 -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.9.0/index.html
```
6.安装MMdetection:可以使用以下命令安装。
```
git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
pip install -r requirements/build.txt
pip install -v -e . # or "python setup.py develop"
```
7.测试:可以使用以下命令测试是否安装成功。
```
python tools/train.py configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py
```