【问题标题】:Run docker nvidia container with GPUS and systemctl使用 GPUS 和 systemctl 运行 docker nvidia 容器
【发布时间】:2020-12-03 11:02:19
【问题描述】:
当尝试将 systemctl 与 GPU 一起运行时,
如果没有 --privileged,systemctl 将无法工作,并且当尝试通过提供 --runtime=nvidia -e NVIDIA_VISIBLE_DEVICES=0 --privileged 来限制 GPUS 时,容器会显示容器内所有可用的 GPU。
如果未指定 --privileged --runtime=nvidia -e NVIDIA_VISIBLE_DEVICES=0,容器会显示所需的 GPU 数量
【问题讨论】:
标签:
docker
containers
nvidia-docker
【解决方案1】:
NV_GPU='GPU 编号' nvidia-docker run --runtime=nvidia -it -v path:path nvcr.io/nvidia/tensorflow:xx.xx
例子:
NV_GPU='0,1' nvidia-docker run --runtime=nvidia -it -v /storage/research/:/storage/research/ nvcr.io/nvidia/tensorflow:17.11-tf2-p3