docker-compose-gpu.yml

version: '3.7'
services:
  workflow:
    image:
      context: .
      dockerfile: ./deploy/Dockerfile
    command: ["jupyter", "lab", "--no-browser", "--allow-root", "--ip", "0.0.0.0", "--port", "8888"]
    ports:
      - 8888:8888
    volumes:
      - './data:/data/'
      - './deploy:/workdir'
    working_dir: /workdir
    networks:
      app_subnet:
        ipv4_address: 172.20.0.10
    depends_on:
      - redisai

  redisai:
      image: 'redislabs/redisai:1.2.7-bionic-gpu'
      ports:
        - '6379:6379'
      volumes:
        - redisai_data:/var/lib/redis
        - ./config/redis.conf:/usr/local/etc/redis/redis.conf
      networks:
        app_subnet:
          ipv4_address: 172.20.0.20
    deploy:
      resources:
        reservations:
          devices:
            - driver: "nvidia"
              capabilities: ["gpu"]
              count: 1

volumes:
  workflow:
  redisai_data:
networks:
  app_subnet:
    driver: bridge
    ipam:
      config:
        - subnet: 172.20.0.0/24