在Docker启动时,显式地声明挂载GPU设备,即使用--gpus参数来指定可见的GPU设备。例如:
--gpus
docker run --gpus all -it ubuntu bash
这样,容器内部就能够访问到宿主机上的所有GPU设备了。如果只想挂载部分GPU设备,可以使用--gpus参数后面跟随的设备ID列表来实现,例如:
docker run --gpus device=0,3 -it ubuntu bash
这样,容器内部就只能访问到宿主机上ID为0和3的GPU设备了。
上一篇:标准幻灯片操作后返回“错误响应400“请求包含无效参数”。
下一篇:标准化Numpy数组