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

feature: How to use multi GPU #5233

Open
Euraxluo opened this issue Feb 20, 2025 · 3 comments
Open

feature: How to use multi GPU #5233

Euraxluo opened this issue Feb 20, 2025 · 3 comments
Labels
enhancement Enhancement proposals

Comments

@Euraxluo
Copy link

Feature request

An example of a model that uses multiple gpus at the same time, or we should have a way to make this easy to use

Motivation

No response

Other

No response

@Euraxluo Euraxluo added the enhancement Enhancement proposals label Feb 20, 2025
@frostming
Copy link
Collaborator

@bentoml.service(resources={"gpu": 4})
class MyService:
    ...

And it will inject CUDA_VISIBLE_DEVICES=0,1,2,3 into the environment

@Euraxluo
Copy link
Author

Will it automatically schedule and use all gpus at the same time?

@frostming
Copy link
Collaborator

What automation do you mean, that is all it does and it depends on how the framework respects the env var.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement Enhancement proposals
Projects
None yet
Development

No branches or pull requests

2 participants