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]: How to specify a GPU when i use vllm by 'python -m vllm.entrypoints.openai.api_server --served-model-name..' #5585

Closed
GoogleAlphaZero opened this issue Jun 17, 2024 · 2 comments
Labels
usage How to use vllm

Comments

@GoogleAlphaZero
Copy link

Your current environment

The output of `python collect_env.py`

How would you like to use vllm

I want to run inference of a [specific model](put link here). I don't know how to integrate it with vllm.

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

You can use CUDA_VISIBLE_DEVICES environment variable to limit the GPUs that can be used by vLLM.

@GoogleAlphaZero
Copy link
Author

GoogleAlphaZero commented Jun 17, 2024 via email

@hmellor hmellor closed this as completed Jul 4, 2024
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

3 participants