dockerfiles-deep-learning

Docker for Deep Learning with Keras/Tensorflow

Install

Install Docker

https://docs.docker.com/engine/installation/

Install NVIDIA driver

Add repo and install

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
sudo apt-get install nvidia-378

Reboot system

sudo shutdown -r now

Check driver version

cat /proc/driver/nvidia/version

Install NVIDIA-Docker

https://github.com/NVIDIA/nvidia-docker

Build

...

Run

...