master_thesis_models / docker-compose.yml
Hannes Kuchelmeister
change docker-compose to not use gpu
993f17b
raw
history blame
473 Bytes
version: "3"
services:
model:
build:
context: .
dockerfile: Dockerfile
tty: true
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=0"
tensorboard:
image: tensorflow/tensorflow
volumes:
- ./logs:/logs:z
command: sh -c "tensorboard --logdir /logs --bind_all"
ports:
- 8008:6006