NVIDIA-GPU-Tensor-Core-Accelerator-PyTorch-OpenCV/docker-compose.yaml at master · zacpullen/NVIDIA-GPU-Tensor-Core-Accelerator-PyTorch-OpenCV · 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
35
36
37
38
version: '2.3'

services:
  computer-vision:
    build:
      context: .
      dockerfile: Dockerfile
    runtime: nvidia
    ports:
      - '8888:8888'
      - '6006:6006'
    volumes:
      - ./app:/app
    environment:
      - NVIDIA_VISIBLE_DEVICES=all
    networks:
      - computer_vision_api_bridge

  tensorflow-serving:
    image: 'tensorflow/serving'
    ports:
      - '8500:8500'
      - '8501:8501'
    volumes:
      - ./models:/models
    environment:
      - MODEL_NAME=fake_model
      - NVIDIA_VISIBLE_DEVICES=all
    depends_on:
      - computer-vision

networks:
  computer_vision_api_bridge:
    driver: bridge

volumes:
  app:
  model_serving: