fly-netsphere/docker-compose.yml at main · bitdeep/fly-netsphere · 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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
services:
  fly:
    profiles: ["recording"]
    build: .
    working_dir: /app
    volumes:
      - .:/app
      - warp-cache:/root/.cache/warp
    environment:
      MUJOCO_GL: egl
      MUJOCO_EGL_DEVICE_ID: "0"
      NVIDIA_VISIBLE_DEVICES: all
      NVIDIA_DRIVER_CAPABILITIES: graphics,compute,utility
      TF_CPP_MIN_LOG_LEVEL: "2"
      PYTHONPATH: /app/flybody
      OPENBLAS_NUM_THREADS: "1"
      OMP_NUM_THREADS: "1"
      TF_NUM_INTRAOP_THREADS: "1"
      TF_NUM_INTEROP_THREADS: "1"
    gpus: all
    command: bash scripts/simulate_city.sh

  neural-tools:
    profiles: ["neural-tools"]
    build:
      context: .
      dockerfile: Dockerfile.neural-tools
    image: fly-neural-tools:0.1.0
    working_dir: /app
    user: "${LOCAL_UID:-1000}:${LOCAL_GID:-1000}"
    volumes:
      - .:/app:ro
      - ./data/neural-reference:/data
      - ./out/neural-reference:/out
    environment:
      MUJOCO_GL: disable
      OPENBLAS_NUM_THREADS: "1"
      OMP_NUM_THREADS: "1"
    init: true
    read_only: true
    tmpfs:
      - /tmp:size=32m,mode=1777
    cap_drop: [ALL]
    security_opt: ["no-new-privileges:true"]
    mem_limit: 512m
    cpus: 0.5
    pids_limit: 32

  environment-dev:
    profiles: ["dev"]
    build: .
    image: fly-environment:0.1.0
    working_dir: /app
    command: python scripts/dev_environment.py --port 8089
    volumes:
      - .:/app:ro
    ports:
      - "127.0.0.1:8089:8089"
    environment:
      MUJOCO_GL: disable
      OPENBLAS_NUM_THREADS: "1"
      OMP_NUM_THREADS: "1"
      FLY_DEV: "1"
      FLY_NEURAL: "1"
    init: true
    read_only: true
    cap_drop: [ALL]
    security_opt: ["no-new-privileges:true"]
    mem_limit: 1g
    cpus: 2.0
    pids_limit: 64
    healthcheck:
      test: ["CMD", "python", "-c", "import urllib.request; urllib.request.urlopen('http://127.0.0.1:8089/health', timeout=3)"]
      interval: 15s
      timeout: 5s
      retries: 3
      start_period: 30s

volumes:
  warp-cache: