It is called the NVIDIA Container Toolkit! Nvidia Container Toolkit ( Citation) Potential Errors in Docker Luckily, you have found the solution explained here. Certain things like the CPU drivers are pre-configured for you, but the GPU is not configured when you run a docker container. To add another layer of difficulty, when Docker starts a container - it starts from almost scratch. The configuration steps change based on your machine's operating system and the kind of NVIDIA GPU that your machine has. In this post, we walk through the steps required to access your machine's GPU within a Docker container.Ĭonfiguring the GPU on your machine can be immensely difficult.
