Docker 中的 GPU 访问可让您将要求苛刻的工作负载(例如机器学习应用程序)容器化。当您启动新容器时,GPU 不会自动可用,但可以使用–gpus标志docker run或通过向文件添加额外字段来激活它们docker-compose.yml。
在本文中,我们将展示如何在 Docker Compose 中启用 GPU 支持。您需要 Docker Compose 版本 v1.28 或更高版本才能遵循该指南。Compose 版本 v1.18 及更早版本不支持 GPU;v1.19 和 v1.27 之间的版本使用提供较少控制的遗留字段结构。
准备您的系统
您的 Docker 主机需要先准备好,然后才能公开您的 GPU 硬件。尽管容器共享您主机的内核,但它们看不到您已安装的系统包。普通容器将缺少与 GPU 交互的设备驱动程序。
您可以通过安装 NVIDIA 的 Docker Container Toolkit 来激活对 NVIDIA GPU 的支持:
distribution=