GPU support for Docker
You can use GPUs in big data applications such as machine learning, data analytics, and genome sequencing. Docker containerization makes it easier for you to package and distribute applications. You can enable GPU support when using YARN on Docker containers.
The NVIDIA Docker plug-in enables you to deploy GPU-accelerated applications with NVIDIA Docker support. When you use Docker containers, you must install NVIDIA Docker plug-in 1.0 to detect and set up GPU containers automatically.