【问题标题】: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
    

    【讨论】:

      猜你喜欢
      • 2020-10-28
      • 1970-01-01
      • 2019-03-06
      • 1970-01-01
      • 1970-01-01
      • 2021-10-20
      • 2018-03-29
      • 2016-07-10
      • 2019-02-03
      相关资源
      最近更新 更多