entity/docker-compose.yml at master · StaticObserver/entity · 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
x-common-config: &common-config
  stdin_open: true
  tty: true
  ports:
    - "8080:8080"
  volumes:
    - type: bind
      source: .
      target: /root/entity/

x-cuda-base: &cuda-base
  image: morninbru/entity:cuda

x-rocm-base: &rocm-base
  image: morninbru/entity:rocm

services:
  entity-cuda-compilers:
    container_name: entity_cuda_compilers
    <<: [*cuda-base, *common-config]
  entity-cuda:
    container_name: entity_cuda
    runtime: nvidia
    environment:
      - NVIDIA_VISIBLE_DEVICES=all
    <<: [*cuda-base, *common-config]
  entity-rocm-compilers:
    container_name: entity_rocm_compilers
    <<: [*common-config, *rocm-base]
  entity-rocm:
    container_name: entity_rocm
    devices:
      - "/dev/kfd"
      - "/dev/dri"
    security_opt:
      - seccomp:unconfined
    <<: [*common-config, *rocm-base]