Public | Automated Build

Last pushed: a year ago
Short Description
Cuda + Theano + Conda + Python 3 + Keras
Full Description

dockerfile-cuda-keras

===========
Ubuntu Core 14.04 + CUDA 7.5.18 + cuDNN v4 + Conda + Python 3 + Theano + Keras.

Requirements

  • Host with corresponding CUDA drivers (v. 352.39) installed for the kernel module.

Usage

The container must have all NVIDIA devices attached to it for CUDA to work properly.
Therefore the command will be as such: docker run -it --device /dev/nvidiactl --device /dev/nvidia-uvm --device /dev/nvidia0 kaixhin/cuda-theano.
With 4 GPUs this would also have to include --device /dev/nvidia1 --device /dev/nvidia2 --device /dev/nvidia3.

For more information on CUDA on Docker, see the repo readme.

Citation

If you find this useful in research please consider citing this work.

Docker Pull Command
Owner
lukovkin
Source Repository