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.