mirror of
https://github.com/deepseek-ai/DeepSeek-Prover-V1.5.git
synced 2026-11-05 00:57:20 +00:00
[GH-ISSUE #11] Seems that all models are loaded to the first GPU? #8
Labels
No labels
pull-request
No milestone
No project
No assignees
1 participant
Notifications
Due date
No due date set.
Dependencies
No dependencies set.
Reference
deepseek-ai/DeepSeek-Prover-V1.5#8
Loading…
Add table
Add a link
Reference in a new issue
No description provided.
Delete branch "%!s()"
Deleting a branch is permanent. Although the deleted branch may continue to exist for a short time before it actually gets removed, it CANNOT be undone in most cases. Continue?
Originally created by @DDDOH on GitHub (Nov 25, 2024).
Original GitHub issue: https://github.com/deepseek-ai/DeepSeek-Prover-V1.5/issues/11
I use four A100 run
python -m prover.launch --config=configs/RMaxTS.py --log_dir=logs/RMaxTS_resultsBut always get cuda out of memory error.
I added some print message:

Then i get

All models are tried to load in first GPU. Is this expected? Or maybe pytorch behavior is different for new version (I am using '2.5.1+cu121', and the version in requirement.txt is 2.2.1)
@Luobots commented on GitHub (Dec 15, 2024):
Same error. VLLM can't set the cuda index.
@zzhisthebest commented on GitHub (Jan 4, 2025):
@Luobots How to solve this problem?I meet the same error.
@chuxuecao commented on GitHub (Apr 24, 2025):
@zzhisthebest Hi, have you solved the problem?