docker compose + 딥러닝 도커 환경에서 웹캠 사용하기

yun·2023년 11월 23일
1

Docker

목록 보기
5/7

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

0개의 댓글