在nvidia-docker中限制GPU的使用?

我正在多GPU服务器上设置一个内部Jupyterhub。 Jupyter访问通过docker实例提供。 我想限制每个用户的访问不超过一个GPU。 我会很感激任何build议或意见。 谢谢。

你可以用nvidia-docker-compose来试试

version: "2" services process1: image: nvidia/cuda devices: - /dev/nvidia0