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

How to run a docker container with specific GPUs using Docker SDK for Python

在命令行中,我习惯于使用 --gpus 参数 run/create 具有特定 GPU 的容器:

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

用于 Python 文档的 Docker SDK 不是很有帮助,我找不到关于如何使用 python SDK 执行相同操作的很好的解释。有办法吗?

这就是您如何 run/create docker 具有特定 GPU 的容器,类似于 --gpu 参数:

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/