- Host with corresponding CUDA drivers (v. 352.39) installed for the kernel module.
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.
If you find this useful in research please consider citing this work.