InternLM / InternLM/xtuner

单卡多机无法训练千问2.5

Open
#970 3 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
5.2k
Forks
448
Avg merge
3d 15h
Merged PRs (30d)
26

Description

## 运行指令:
```
CUDA_VISIBLE_DEVICES=1,2,3,4 NPROC_PER_NODE=4 xtuner train ./internlm2_chat_1_8b_dpo_full_copy.py
```

## internlm2_chat_1_8b_dpo_full_copy.py
这里我在示例的基础上把数据集和模型改成了千问2.5_32b的模型

## 运行部分结果:

```python

W1209 06:53:34.745000 139752507041600 torch/distributed/elastic/multiprocessing/api.py:858] Sending process 34366 closing signal SIGTERM
W1209 06:53:34.747000 139752507041600 torch/distributed/elastic/multiprocessing/api.py:858] Sending process 34367 closing signal SIGTERM
W1209 06:53:34.747000 139752507041600 torch/distributed/elastic/multiprocessing/api.py:858] Sending process 34368 closing signal SIGTERM
E1209 06:53:46.195000 139752507041600 torch/distributed/elastic/multiprocessing/api.py:833] failed (exitcode: -9) local_rank: 3 (pid: 34369) of binary: /mnt/public/conda/envs/xtuner/bin/python
Traceback (most recent call last):
File "/mnt/public/conda/envs/xtuner/bin/torchrun", line 8, in
sys.exit(main())
File "/mnt/public/conda/envs/xtuner/lib/python3.10/site-packages/torch/distributed/elastic/multiprocessing/errors/__init__.py", line 348, in wrapper
return f(*args, **kwargs)
File "/mnt/public/conda/envs/xtuner/lib/python3.10/site-packages/torch/distributed/run.py", line 901, in main
run(args)
File "/mnt/public/conda/envs/xtuner/lib/python3.10/site-packages/torch/distributed/run.py", line 892, in run
elastic_launch(
File "/mnt/public/conda/envs/xtuner/lib/python3.10/site-packages/torch/distributed/launcher/api.py", line 133, in __call__
return launch_agent(self._config, self._entrypoint, list(args))
File "/mnt/public/conda/envs/xtuner/lib/python3.10/site-packages/torch/distributed/launcher/api.py", line 264, in launch_agent
raise ChildFailedError(
torch.distributed.elastic.multiprocessing.errors.ChildFailedError:
============================================================
/mnt/public/conda/envs/xtuner/lib/python3.10/site-packages/xtuner/tools/train.py FAILED
------------------------------------------------------------
Failures:

------------------------------------------------------------
Root Cause (first observed failure):
[0]:
time : 2024-12-09_06:53:34
host : is-dahisl6olik7jjio-devmachine-0
rank : 3 (local_rank: 3)
exitcode : -9 (pid: 34369)
error_file:
traceback : Signal 9 (SIGKILL) received by PID 34369
============================================================

```

## 输出的日志文件出错部分内容:

```
2024/12/09 02:59:17 - mmengine - WARNING - Failed to search registry with scope "mmengine" in the "builder" registry tree. As a workaround, the current "builder" registry in "xtuner" is used to build instance. This may cause unexpected failure when running the built modules. Please check whether "mmengine" is a correct scope, or whether the registry is initialized.

```

## 尝试了的方法

1. 将max_length改成了1024和512仍然不行,不知道是不是OOM,但是我试过跑6张A100 80G的卡都是同样的问题
2. 跑单卡,单卡直接报错内存不足,没有抛出上面`torch.distributed.elastic.multiprocessing.errors.ChildFailedError: `的信息(其实也合理,毕竟没有使用多卡,也就没有分布式一说了)

现在我怀疑是不是XTuner不支持Qwen2.5了,看了仓库其他Issue似乎都没有跟我类似的问题,而之前有人问支不支持qwen也没有人回复。。。
求各位大佬帮帮忙!!!

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.