How to Use an NVIDIA GPU with Docker Containers
Docker containers don’t see your system’s GPU automatically. This causes reduced performance in GPU-dependent workloads such as machine learning frameworks. Here’s how to expose your host’s NVIDIA GPU to your containers.
Source: howtogeek.com