merge compose files

docker-compose.cuda.yml

@@ -1,29 +0,0 @@
-version: "3"
-services:
- model-cuda:
- env_file:
- - .env 
- user: "${UID}:${GID}"
- build: 
- context: .
- dockerfile: DockerfileCUDA
- tty: true
- deploy:
- resources:
- reservations:
- devices:
- - driver: nvidia
- count: 1
- capabilities: [gpu]
- volumes:
- - ./logs/:/usr/src/app/logs:z
- - ./data/:/usr/src/app/data:z
- - ./configs:/usr/src/app/configs:z
- command: sh -c "python train.py logger=tensorboard trainer.gpus=1"
- tensorboard:
- image: tensorflow/tensorflow
- volumes:
- - ./logs:/logs:z
- command: sh -c "tensorboard --logdir /logs --bind_all"
- ports:
- - 8008:6006

docker-compose.yml

@@ -1,5 +1,26 @@
 version: "3"
 services:
+ model-cuda:
+ env_file:
+ - .env
+ user: "${UID}:${GID}"
+ build:
+ context: .
+ dockerfile: DockerfileCUDA
+ tty: true
+ deploy:
+ resources:
+ reservations:
+ devices:
+ - driver: nvidia
+ count: 1
+ capabilities: [gpu]
+ volumes:
+ - ./logs/:/usr/src/app/logs:z
+ - ./data/:/usr/src/app/data:z
+ - ./configs:/usr/src/app/configs:z
+ command: sh -c "python train.py logger=tensorboard trainer.gpus=1"
+
  model:
  env_file:
  - .env
@@ -13,6 +34,7 @@ services:
  - ./data/:/usr/src/app/data:z
  - ./configs:/usr/src/app/configs:z
  command: sh -c "python train.py logger=tensorboard trainer.gpus=0"
+
  tensorboard:
  image: tensorflow/tensorflow
  volumes: