如何使用 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/
在命令行中,我习惯于使用 --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/