docker-compose.yml

version: '2'

volumes:
  nvidia:
    external:
      name: nvidia_driver_367.57
  shared:

services:
  tf-cpu:
    build:
      context: ./docker
      dockerfile: cpu.dockerfile
    volumes:
      - ./notebooks:/notebooks
      - ./data:/data
      - ./models/:/models
      - ./:/code
      - shared:/logs
      - ../tensorbuilder:/tensorbuilder
    ports:
      - "8885:8888"

  tf-gpu:
    build:
      context: ./docker
      dockerfile: gpu.dockerfile
    image: tf-gpu
    volumes:
      - ./notebooks:/notebooks
      - ./data:/data
      - ./models/:/models
      - ./:/code
      - shared:/logs
      - nvidia:/usr/local/nvidia:ro
    devices:
      - /dev/nvidia0:/dev/nvidia0
      - /dev/nvidiactl:/dev/nvidiactl
      - /dev/nvidia-uvm:/dev/nvidia-uvm
    ports:
      - "8886:8888"

  tensorboard:
    build:
      context: ./docker
      dockerfile: cpu.dockerfile
    volumes:
      - shared:/logs
    ports:
      - "6006:6006"
    command: bash -c "rm -fr /logs/* && tensorboard --logdir=/logs"