时间戳

首页 > 解决方案 > 用于 Python 的 Nvidia-Docker API？

python - 用于 Python 的 Nvidia-Docker API？

问题描述

我目前正在运行许多类似的 Docker 容器，这些容器是由 Python 脚本通过官方 API 创建和运行的。由于 Docker 本身不支持 GPU 映射，因此我测试了 Nvidia-Docker，它满足了我的要求，但我不确定如何将它无缝集成到我的脚本中。

我尝试使用 Google 和文档为 Nvidia-Docker 找到正确的 API 调用，但我没有找到任何有用的东西。

我当前的代码如下所示：

# assemble a new container using the params obtained earlier
container_id = client.create_container(img_id, command=commands, stdin_open=True, tty=True, volumes=[folder], host_config=client.create_host_config(binds=[mountpoint,]),detach=False) 
# run it 
client.start(container_id)

API 的文档可以在这里找到。

来自Nvidia-Dockers Github页面：

Docker® 引擎使用的默认运行时是 runc，我们的运行时可以通过使用 --default-runtime=nvidia 配置 docker 守护进程成为默认运行时。这样做将无需将 --runtime=nvidia 参数添加到 docker run。这也是在 docker build 期间获得 GPU 访问权限的唯一方法。

基本上，我想将 --runtime=nvidia-docker 参数添加到我的 create_container 调用中，但似乎不支持它。

但是由于我需要在脚本执行期间多次在运行时之间切换（混合 Nvidia-Docker 和本机 Docker 容器），快速而肮脏的方法是使用子进程运行 bash 命令，但我觉得必须有更好的方法。

TL;DR：我正在寻找一种从 Python 脚本运行 Nvidia-Docker 容器的方法。

标签： pythondockernvidia-docker

解决方案

run()并且create()方法具有runtime根据https://docker-py.readthedocs.io/en/stable/containers.html的参数

这很有意义，因为dockercli 工具非常简单，并且每个命令都在调用 docker 引擎服务 REST API 时进行转换

推荐阅读