Retratista/docker-compose.yml at main · DarkGeekMS/Retratista · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
version: "2.3"

services:
    stylegan_server:
        image: retratista:stylegan_server
        build:
            context: .
            dockerfile: ./docker/stylegan_server/Dockerfile
        runtime: nvidia
        ports:
            - "5000:5000"
        environment:
            FLASK_ENV: production
            NVIDIA_VISIBLE_DEVICES: all
    ui:
        image: retratista:ui
        build:
            context: .
            dockerfile: ./docker/ui/Dockerfile
        ports:
            - "8080:8080"
        environment:
            TYPE: $TYPE_ARG
    pose_server:
        image: retratista:pose_server
        build:
            context: .
            dockerfile: ./docker/pose_server/Dockerfile
        runtime: nvidia
        ports:
            - "5001:5001"
        environment:
            FLASK_ENV: production
            NVIDIA_VISIBLE_DEVICES: all