- Python 3.5
- Keras 2
- Tensorflow 1.1.0
- scipy, numpy, scikit-learn
- hdf5, h5py
- Docker Engine
Docker host requirements. (made for AWS p2 instance)
- GPU (NVIDIA Tesla K80)
- CUDA Toolkit / CUDA Driver
- Docker Engine
- nvidia-docker
$ python3 cli.py
or execute on a Docker container,
$ NOGPU=1 ./run
$ make build
$ make build-nogpu