alibaba / alibaba/ROLL

QwenVL Training Error When Using Specific GPUs on a Single Server

Open
#78 2 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
3.4k
Forks
312
Avg merge
1h 2m
Merged PRs (30d)
2

Description

I want to train the QwenVL model using GPU 2 and GPU 3 on a single server. I modified the YAML configuration to use two GPUs and set the device mapping accordingly. At the same time, I specified CUDA_VISIBLE_DEVICES as 2 and 3 in the shell script.

However, I encountered the following problems:

1. The program still runs on GPU 0 and GPU 1, not on GPU 2 and GPU 3 as intended.

2. The inference processes (actor_infer-0 and actor_infer-1) both report errors related to CUDA, saying there is an "invalid argument" and failing during collective RPC.

3. Meanwhile, the training processes (actor_train-0 and actor_train-1) do not report any error and run normally.

Could you please help me understand why:

The inference actors are failing;

The training still uses GPU 0 and 1 even though I tried to assign GPU 2 and 3?

Thanks a lot!

[yaml.txt](https://github.com/user-attachments/files/21013280/yaml.txt)
![Image](https://github.com/user-attachments/assets/f8ade8b4-75df-4270-9b64-4ae1e3e618e8)
![Image](https://github.com/user-attachments/assets/24d211d9-a4fa-49bf-b826-0fb571799dfb)

Contributor guide

No contributing guide indexed for this repository

Research direction

Start with the attached yaml.txt and the shell script that sets CUDA_VISIBLE_DEVICES=2,3, then inspect the actor_infer and actor_train logs. Reproduce the single-server two-GPU configuration and compare logical device numbering with the configured device mapping. Done means explaining the inference CUDA invalid-argument failure and why training selects GPUs 0 and 1, with a verified configuration or documented limitation.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
distributed-systems, machine-learning
Issue type
Bug
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.