RAPIDS Docker Run Command Changes in v22.04

Overview

As we expand out further across platforms and GPU architectures, we have recently identified and updated the RAPIDS docker run with additional parameters to provide a more stable way to run Multi Node/Multi GPU (MNMG). The new parameters are --shm-size=1g --ulimit memlock=-1. This change is currently in the RAPIDS Release Selector and will soon be updated in all docker instructions. We advise everyone to adopt these additional parameters for when running RAPIDS on Docker, especially if you’re running RAPIDS on ARM SBSA with MNMG.

Status

  • 2022-06-08 - This change is currently in the RAPIDS Release Selector

Impact

Refer to the examples below for a comparison of old & new docker run commands:

Example Old Docker Run Command

    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

Example New Docker Run Command

    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