Skip to content

Latest commit

 

History

History
18 lines (11 loc) · 320 Bytes

README.md

File metadata and controls

18 lines (11 loc) · 320 Bytes

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 &