Using the webcam in a GPU enabled Docker Compose

camera.py

import cv2

video_capture = cv2.VideoCapture(0)
anterior = 0

while True:
    if not video_capture.isOpened():
        print('Unable to load camera. Use the command "xhost +"')
        pass

    # Capture frame-by-frame
    ret, frame = video_capture.read()

    # Display the resulting frame
    cv2.imshow('Video', frame)


    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

    # Display the resulting frame
    cv2.imshow('Video', frame)

# When everything is done, release the capture
video_capture.release()
cv2.destroyAllWindows()

docker-compose.yml

version: '2.3'

services:

  webcam:
    image: webcam
    container_name: webcam
    working_dir: /webcam
    build:
      context: .
      dockerfile: Dockerfile.webcam
    environment:
      - DISPLAY=unix$DISPLAY
    volumes:
      - ./:/webcam
      - /dev/video0:/dev/video0
      - /tmp/.X11-unix:/tmp/.X11-unix
    command: bash -c "nvidia-smi && python -u camera.py"
    privileged: true
    runtime: nvidia

Dockerfile.webcam

FROM tensorflow/tensorflow:latest-gpu-py3

RUN apt update -y && apt install -y \
libsm6 \
libxext6 \
libxrender-dev

RUN pip install \
opencv-python