Hannes Kuchelmeister commited on
Commit
34a2150
1 Parent(s): 28843d2

merge compose files

Browse files
Files changed (2) hide show
  1. docker-compose.cuda.yml +0 -29
  2. docker-compose.yml +22 -0
docker-compose.cuda.yml DELETED
@@ -1,29 +0,0 @@
1
- version: "3"
2
- services:
3
- model-cuda:
4
- env_file:
5
- - .env
6
- user: "${UID}:${GID}"
7
- build:
8
- context: .
9
- dockerfile: DockerfileCUDA
10
- tty: true
11
- deploy:
12
- resources:
13
- reservations:
14
- devices:
15
- - driver: nvidia
16
- count: 1
17
- capabilities: [gpu]
18
- volumes:
19
- - ./logs/:/usr/src/app/logs:z
20
- - ./data/:/usr/src/app/data:z
21
- - ./configs:/usr/src/app/configs:z
22
- command: sh -c "python train.py logger=tensorboard trainer.gpus=1"
23
- tensorboard:
24
- image: tensorflow/tensorflow
25
- volumes:
26
- - ./logs:/logs:z
27
- command: sh -c "tensorboard --logdir /logs --bind_all"
28
- ports:
29
- - 8008:6006
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
docker-compose.yml CHANGED
@@ -1,5 +1,26 @@
1
  version: "3"
2
  services:
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
3
  model:
4
  env_file:
5
  - .env
@@ -13,6 +34,7 @@ services:
13
  - ./data/:/usr/src/app/data:z
14
  - ./configs:/usr/src/app/configs:z
15
  command: sh -c "python train.py logger=tensorboard trainer.gpus=0"
 
16
  tensorboard:
17
  image: tensorflow/tensorflow
18
  volumes:
 
1
  version: "3"
2
  services:
3
+ model-cuda:
4
+ env_file:
5
+ - .env
6
+ user: "${UID}:${GID}"
7
+ build:
8
+ context: .
9
+ dockerfile: DockerfileCUDA
10
+ tty: true
11
+ deploy:
12
+ resources:
13
+ reservations:
14
+ devices:
15
+ - driver: nvidia
16
+ count: 1
17
+ capabilities: [gpu]
18
+ volumes:
19
+ - ./logs/:/usr/src/app/logs:z
20
+ - ./data/:/usr/src/app/data:z
21
+ - ./configs:/usr/src/app/configs:z
22
+ command: sh -c "python train.py logger=tensorboard trainer.gpus=1"
23
+
24
  model:
25
  env_file:
26
  - .env
 
34
  - ./data/:/usr/src/app/data:z
35
  - ./configs:/usr/src/app/configs:z
36
  command: sh -c "python train.py logger=tensorboard trainer.gpus=0"
37
+
38
  tensorboard:
39
  image: tensorflow/tensorflow
40
  volumes: