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.

Sign up for GitHub

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

Jump to bottom

[Feature] Support CUDA_VISIBLE_DEVICES and multiple tasks on one GPU #148

Merged
merged 3 commits into from
Aug 10, 2023
Merged

[Feature] Support CUDA_VISIBLE_DEVICES and multiple tasks on one GPU #148

merged 3 commits into from
Aug 10, 2023

Conversation

mzr1996
Copy link
Collaborator

@mzr1996 mzr1996 commented Aug 2, 2023

Closing #55

gaotongxiao
gaotongxiao reviewed Aug 2, 2023
View reviewed changes
Copy link
Collaborator

@gaotongxiao gaotongxiao left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

We need to update run.py to include max_workers_per_gpu as well; then update the docs in another PR

opencompass/runners/local.py Outdated Show resolved Hide resolved
@gaotongxiao gaotongxiao linked an issue Aug 10, 2023 that may be closed by this pull request
[Bug] 无法指定gpu #55
Closed
@gaotongxiao gaotongxiao merged commit 59bf563 into main Aug 10, 2023
2 checks passed
@gaotongxiao gaotongxiao deleted the mzr/update-local branch August 10, 2023 08:53
go-with-me000 pushed a commit to go-with-me000/opencompass that referenced this pull request Oct 9, 2023
@mzr1996 @BunnyRunnerX
Support CUDA_VISIBLE_DEVICES and multiple tasks on one GPU (open-comp…
2b69da9 
…ass#148)

* [Feature] Support CUDA_VISIBLE_DEVICES and multiple tasks on one GPU

* Fix UT

* Update according to comments
liuyaox pushed a commit to liuyaox/opencompass that referenced this pull request Jun 26, 2024
@mzr1996
[Feature] Support CUDA_VISIBLE_DEVICES and multiple tasks on one GPU (o…
7f2faa7 
…pen-compass#148)

* [Feature] Support CUDA_VISIBLE_DEVICES and multiple tasks on one GPU

* Fix UT

* Update according to comments
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@gaotongxiao gaotongxiao gaotongxiao approved these changes

Assignees

@yingfhu yingfhu

Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.

[Bug] 无法指定gpu
3 participants
@mzr1996 @gaotongxiao @yingfhu