docker compose + 딥러닝 도커 환경에서 웹캠 사용하기
Dockerfile
- YOLOv8과 torch, tensorflow, mediapipe를 통해 학습
- opencv를 통해 학습된 모델로 객체 인식
- root가 아닌 일반 사용자로 접속
compose.yaml
- 사용자 환경에 맞게 USER_NAME과 volume은 변경 필요
- host와 같은 network를 사용하게 했으나, 명시하지 않아도 bridge로 컨테이너 간에는 동일한 네트워크 위에 있게 됨. 필요 시 수정할 것.
- 웹캠과 GPU를 사용할 수 있게 함
- 실행 후 종료될 경우 자동 재시작
실행 방법
docker compose up -d
컨테이너를 삭제해야 한다면
- compose.yaml 파일에서 restart always로 되어 있는 경우 컨테이너를 삭제하더라도 계속 재시작됨
- docker stop <container_name>으로 stop 후 rm