[MMDetection] 환경설정

1. MMDetection 설치 및 실행

- pytorch cuda버전별 설치:  https://pytorch.org/get-started/previous-versions/

- mmdet cuda버전별 설치: https://mmdetection.readthedocs.io/en/latest/get_started.html#customize-installation

- 운영체제와 CUDA버전에 맞는 pytorch 설치

# CUDA 11.6
pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116

 

- MIM을 이용하여 MMEngine과 MMCV설치 (mmcv-full이 있는 경우 제거후 MMCV 설치)

# openmim 설치
pip install -U openmim

# mim을 이용한 mmengine 설치
mim install mmengine    (=pip install mmengine)

# mmcv 존재 여부 검색
pip list | grep mm

# mmcv-full제거
pip uninstall mmcv-full

# mmcv설치(존재시 설치x)
mim install "mmcv>=2.0.0"

# mmcv 강제 재설치
mim install --force-reinstall "mmcv>=2.0.0"

 

- mmdet 직접 설치

# mmdet 직접 설치
git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
pip install -v -e .

# mmdet 설치
mim install mmdet

 

- 설치 확인

import torch
import mmcv

print(f"torch version: {torch._version__}")
print(f"GPU: {torch.cuda.is_available()}")
print(f"mmcv version: {mmcv.__version__}")

 

 

2. 버전별 mmcv 설치

- https://mmcv.readthedocs.io/en/latest/get_started/installation.html

- 운영체제, CUDA버전, torch버전 확인

pip install mmcv==2.1.0 -f https://download.openmmlab.com/mmcv/dist/cu116/torch1.12/index.html