我目前正在运行许多类似的 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 的文档可以在这里找到。
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 容器的方法。