How to Use the GPU within a Docker Container
blog.roboflow.com › use-the-gpu-in-dockerMay 18, 2020 · Now we run the container from the image by using the command docker run --gpus all nvidia-test. Keep in mind, we need the --gpus all or else the GPU will not be exposed to the running container. From this base state, you can develop your app accordingly. In my case, I use the NVIDIA Container Toolkit to power experimental deep learning frameworks.
docker run | Docker Documentation
docs.docker.com › engine › referenceThe docker run command first creates a writeable container layer over the specified image, and then starts it using the specified command. That is, docker run is equivalent to the API /containers/create then /containers/ (id)/start. A stopped container can be restarted with all its previous changes intact using docker start.