deepspeedai / deepspeedai/DeepSpeed

[BUG] Cannot run deepspeed in containers

Open
#4,971 6 comments 0 reactions 1 assignee View on GitHub

@loadams is already working on this.

Since Jan 22, 2024.

bug training
Dominant language
Python
Stars
43.1k
Forks
5k
Avg merge
4d 15h
Merged PRs (30d)
112

Description

Describe the bug
Unable to run deepspeed cifar in containerized environments (unless you control the container host) - I think because of numactl?

To Reproduce
Steps to reproduce the behavior:

Start a container, such as NGC. Or go to Google Colab.

git clone deepspeedexamples.

Go to training/cifar

Run run_ds_moe.sh. (If using Colab, adjust the script to set all GPUs/experts vars to 1)

Expected behavior
Runs.

ds_report output

[2024-01-17 19:54:12,758] [INFO] [real_accelerator.py:161:get_accelerator] Setting ds_accelerator to cuda (auto detect)
[2024-01-17 19:54:14,599] [WARNING] [runner.py:202:fetch_hostfile] Unable to find hostfile, will proceed with training with local resources only.
[2024-01-17 19:54:14,605] [INFO] [runner.py:571:main] cmd = /usr/bin/python3 -u -m deepspeed.launcher.launch --world_info=eyJsb2NhbGhvc3QiOiBbMF19 --master_addr=127.0.0.1 --master_port=29500 --enable_each_rank_log=None --bind_cores_to_rank cifar10_deepspeed.py --log-interval 100 --deepspeed --deepspeed_config ds_config.json --moe --ep-world-size 1 --num-experts 1 --top-k 1 --noisy-gate-policy RSample --moe-param-group
[2024-01-17 19:54:17,084] [INFO] [real_accelerator.py:161:get_accelerator] Setting ds_accelerator to cuda (auto detect)
[2024-01-17 19:54:19,709] [INFO] [launch.py:138:main] 0 NV_LIBNCCL_DEV_PACKAGE=libnccl-dev=2.19.3-1+cuda12.2
[2024-01-17 19:54:19,710] [INFO] [launch.py:138:main] 0 NV_LIBNCCL_DEV_PACKAGE_VERSION=2.19.3-1
[2024-01-17 19:54:19,710] [INFO] [launch.py:138:main] 0 NCCL_VERSION=2.19.3-1
[2024-01-17 19:54:19,710] [INFO] [launch.py:138:main] 0 NV_LIBNCCL_DEV_PACKAGE_NAME=libnccl-dev
[2024-01-17 19:54:19,710] [INFO] [launch.py:138:main] 0 NV_LIBNCCL_PACKAGE=libnccl2=2.19.3-1+cuda12.2
[2024-01-17 19:54:19,710] [INFO] [launch.py:138:main] 0 NV_LIBNCCL_PACKAGE_NAME=libnccl2
[2024-01-17 19:54:19,710] [INFO] [launch.py:138:main] 0 NV_LIBNCCL_PACKAGE_VERSION=2.19.3-1
[2024-01-17 19:54:19,710] [INFO] [launch.py:145:main] WORLD INFO DICT: {'localhost': [0]}
[2024-01-17 19:54:19,710] [INFO] [launch.py:151:main] nnodes=1, num_local_procs=1, node_rank=0
[2024-01-17 19:54:19,710] [INFO] [launch.py:162:main] global_rank_mapping=defaultdict(<class 'list'>, {'localhost': [0]})
[2024-01-17 19:54:19,710] [INFO] [launch.py:163:main] dist_world_size=1
[2024-01-17 19:54:19,710] [INFO] [launch.py:165:main] Setting CUDA_VISIBLE_DEVICES=0
set_mempolicy: Operation not permitted
setting membind: Operation not permitted
[2024-01-17 19:54:20,732] [INFO] [launch.py:315:sigkill_handler] Killing subprocess 8319
[2024-01-17 19:54:20,732] [ERROR] [launch.py:321:sigkill_handler] ['numactl', '-m', '0', '-C', '0', '/usr/bin/python3', '-u', 'cifar10_deepspeed.py', '--local_rank=0', '--log-interval', '100', '--deepspeed', '--deepspeed_config', 'ds_config.json', '--moe', '--ep-world-size', '1', '--num-experts', '1', '--top-k', '1', '--noisy-gate-policy', 'RSample', '--moe-param-group'] exits with return code = 1

System info (please complete the following information):

  • OS: Ubuntu 22.04
  • GPU count and types: 1xT4
  • Python version: 3.10.2

Launcher context
Using deepspeed launcher

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.