如何使用 Docker SDK for Python 运行具有特定 GPU 的 Docker 容器

发布于 2025-01-13 02:10:10 字数 193 浏览 2 评论 0原文

在命令行中，我习惯使用 --gpus 参数运行/创建具有特定 GPU 的容器：

docker run -it --gpus '"device=0,2"' ubuntu nvidia-smi

Docker SDK for Python 文档不是很有帮助，而且我找不到关于如何使用 python SDK 执行相同操作的良好解释。有办法做到吗？

原文

分享到QQ

分享到微博

如果你对这篇内容有疑问，欢迎到本站社区发帖提问参与讨论，获取更多帮助，或者扫码二维码加入 Web 技术交流群。

发布评论

需要登录才能够评论，你可以免费注册一个本站的账号。

毁梦 2025-01-20 02:10:10

这是使用 Docker SDK for Python 运行/创建具有特定 GPU 的 docker 容器的方法：

client.containers.run('ubuntu',
                          "nvidia-smi",
                           device_requests=[
                           docker.types.DeviceRequest(device_ids=['0','2'], capabilities=[['gpu']])])

这样您还可以使用此处指定的其他 GPU 资源选项：
https://docs.docker.com/config/containers/resource_constraints/

This is how you can run/create docker containers with specific GPUs using the Docker SDK for Python:

client.containers.run('ubuntu',
                          "nvidia-smi",
                           device_requests=[
                           docker.types.DeviceRequest(device_ids=['0','2'], capabilities=[['gpu']])])

This way you can also use other GPU resource options specified here:
https://docs.docker.com/config/containers/resource_constraints/

回复收藏 0 原文

~没有更多了~