[mmrotate] open-mmlab의 mmrotate docker로 실행하기

이번 포스팅에서는 open-mmlab의 mmrotate를 실행하려 합니다.

우선 git clone을 해주고 폴더 이동을 합니다.

git clone https://github.com/open-mmlab/mmrotate.git
cd mmrotate

참고로 github 주소는 아래와 같습니다.

https://github.com/open-mmlab/mmrotate

GitHub - open-mmlab/mmrotate: OpenMMLab Rotated Object Detection Toolbox and Benchmark

 

 

vi를 이용하여 Dockerfile의 pytorch, cuda, cudnn 버전을 수정할 수 있습니다.(또는 해야할 수 있습니다)

또한 컨테이너 포트 번호도 수정할 수 있습니다.

저는 일단 그대로 사용을 하였습니다.

docker build -t mmrotate docker/

참고로 docker build는 Dockerfile을 image로 만들어주는 명령어입니다.

다운로딩하는데 시간이 조금 걸립니다.

 

 

이미지가 다 생성되었다면 docker를 실행해줍니다.

docker run --gpus all --shm-size=8g -it -p {hostPort}:6006 -p {hostPort}:8888 -v {hostDir}:/mmrotate/data mmrotate

여기서 --shm-size는 공유메모리 크기 설정입니다.

따로 설정하지 않을 시 4M?로 매우 작은 크기이므로 재 설정해줍니다.

-v 는 호스트와 컨테이너 폴더를 연결하며 마찬가지로 -p는 포트를 연결해줍니다.

 

 

 

python에 들어가 mmrotate를 import하고 버전을 프린트합니다.

import mmrotate
print(mmrotate.__version__)

출력:

0.3.2

 

일단 버전까지는 잘 뜨는 것 같습니다.

 

 

 

감사합니다.