Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Usage]: gpu memory usage when using tensor parallel #4880

Open
DaiJianghai opened this issue May 17, 2024 · 1 comment
Open

[Usage]: gpu memory usage when using tensor parallel #4880

DaiJianghai opened this issue May 17, 2024 · 1 comment
Labels
usage How to use vllm

Comments

@DaiJianghai
Copy link

Your current environment

The output of `python collect_env.py`

How would you like to use vllm

i try to use vllm to serve Qwen-32B-chat-AWQ in 3090(24G x 2).
in my expectation, 24G memory could be enough in one gpu, so i use one GPU at first time, but failed
then i try to use tensor parallel to serve the model and that work, but memeory usage over my expection: 18G for each GPU, total 36G, that much more beyond my expectation for one GPU, i want to know, if that is common

in my expectation, 13-14G for each GPU is enough

@DaiJianghai DaiJianghai added the usage How to use vllm label May 17, 2024
@DarkLight1337
Copy link
Collaborator

You can set the --gpu-memory-utilization parameter to a smaller value (default is 90% of each GPU)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
usage How to use vllm
Projects
None yet
Development

No branches or pull requests

2 participants