File size: 638 Bytes
8d9de2d |
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 |
version: "3"
services:
model-cuda:
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
|