Example on how to use multistage builds to build small Docker images using CUDA - View it on GitHub