如何使用 GPU 访问运行 Docker Compose 容器

本文详细介绍了如何在 Docker Compose 中启用 GPU 支持,包括准备系统,确保使用正确的容器镜像,以及通过 `docker-compose.yml` 文件配置 GPU 访问。重点强调了 `deploy.resources.reservations.devices` 字段的使用,允许访问特定数量或特定设备。通过 Docker Compose,可以更方便地管理和使用 GPU 资源。
摘要由CSDN通过智能技术生成

在这里插入图片描述
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=
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

mikes zhang

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值