keras-tf-gpu
Docker image for keras using tensorflow-gpu based on ubuntu 16.04
It includes:
- cuda 8.0
- cudnn 5.0
- tensorflow 1.0
- keras 2.0
Usage
nvidia-docker run --rm -ti xoryouyou/keras-tf-gpu bash
When using tensorboard just put it in the background for example:
tensorboard --logdir=/tmp/logs &