[pytorch] mmdetection 설치 방법

 

torch 사이트에 가면 자신의 cuda 버전과 맞도록 설치할 수 있도록 명령어를 설정해준다.

본인은 cuda 11.8 버전을 사용하고 있으므로 아래와 같이 설치한다.

 

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

 

추천하는 방법은 아래와 같다

pip install openmim
mim install mmdet

 

하지만 나는 이상한 에러들이 너무 많이 떠서 수동을 설치하기로 했다.

 

pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/{cu_version}/{torch_version}/index.html

cuda: 11.8

torch: 2.0.0 이므로 대입해서 설치를 진행했다.

 

그후에 mmdet를 설치했다.

pip install mmdet

 

여기도 종속성 문제로 설치가 제대로 안되긴했는데 어찌됐든 작동은 해서 넘어가기로 했다.