lukovkin/dockerfile-cuda-tensorflow-keras repository overview

⁠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⁠.