[Docker] yolov8 docker 설치해보기

https://github.com/ultralytics/ultralytics

GitHub - ultralytics/ultralytics: YOLOv8 🚀 in PyTorch > ONNX > CoreML > TFLite

 

yolo v5를 만든 팀이 2023년 yolov8를 가지고 돌아왔다.

 

이렇게 쉽게 동작되도 되나 싶을 정도로 쉽게 만들어 졌다.

 

포스팅 하는 기준으로 아직 안정화가 덜 된듯 하지만, 추후 안정화 되면 비전 딥러닝 관련해서 정말 큰 파장(이미 v5도 매우 혁신적이였다.) 

 

을 가져올 거 같다.

 

yolo v8에 대한 소개는 딥러닝 관련 게시물에서 따로 다루고자 한다.

 

오늘은 yolo v8을 도커환경 구축을 포스팅 해보고자 한다.

 

git 주소에 들어가보면 docker를 제공해주고 있다

 

docker 폴더에 들어가보면, 3가지 도커파일을 제공해 주고 있다. (도커 파일에 친절히 설명까지 되어있다.)

Dockerfile -> x86(intel cpu 칩셋)과 Nvidia gpu를 사용하는 장치(보통의 데스크탑 또는 윈도우 & 우분투 노트북)

Dockerfile-arm64 ->  aarch64기반 칩셋인 애플 M1이나 ARM 아키텍쳐들과 Jetson, 라즈베리 파이 등의 장치

Dockerfile-cpu -> cpu만 사용할 경우, ONNX, OpenVINO, Pytorch는 지원이 되는 것 같다.

 

Dockerfile-arm64 파일의 경우, 도커 빌드 중에 (COPY requriement.txt) 부분에서 오류가 난다. 해당 오류는 파악해 봐야 될거같다.

 

설치가 꽤 오래 걸린다. pytorch, opencv-python 등 오래걸리는것들 투성이라 그런거 같다.

 

사이즈도 크다. 15gb 정도니 컴퓨터 용량을 반드시 체크하고 build하도록 하자