RGN 27 - RAPIDS v22.04 Docker Run 命令更改

作者 RAPIDS 运维
状态

已完成

主题 Docker Run 命令更改
RAPIDS 版本 v22.04+
创建日期 2022 年 6 月 8 日
更新日期 2023 年 9 月 5 日

概述

随着我们在平台和 GPU 架构上进一步扩展,我们最近确定并更新了 RAPIDS docker run 命令,增加了额外的参数,以提供更稳定地运行多节点/多 GPU (MNMG) 的方式。新的参数是 --shm-size=1g --ulimit memlock=-1。此更改目前已包含在 RAPIDS 版本选择器中,并将很快更新到所有 docker 说明中。我们建议大家在使用 Docker 运行 RAPIDS 时采用这些额外的参数,特别是当您在 ARM SBSA 上使用 MNMG 运行 RAPIDS 时。

状态

  • 2022-06-08 - 此更改目前已包含在 RAPIDS 版本选择器中

影响

参考以下示例,了解 & docker run 命令的对比。

旧 Docker Run 命令示例

    docker run --gpus all --rm -it \
    -p 8888:8888 -p 8787:8787 -p 8786:8786 \
    rapidsai/rapidsai-core-arm64:22.06-cuda11.5-runtime-ubuntu20.04-py3.9

新 Docker Run 命令示例

    docker run --gpus all --rm -it \
    --shm-size=1g --ulimit memlock=-1 \
    -p 8888:8888 -p 8787:8787 -p 8786:8786 \
    rapidsai/rapidsai-core-arm64:22.06-cuda11.5-runtime-ubuntu20.04-py3.9