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

微调7B的大概用多少GPU #2

Open
tigerbaby1515 opened this issue Apr 12, 2023 · 2 comments
Open

微调7B的大概用多少GPU #2

tigerbaby1515 opened this issue Apr 12, 2023 · 2 comments

Comments

@tigerbaby1515
Copy link

No description provided.

@roy-shih
Copy link

roy-shih commented Jun 2, 2023 •
edited
Loading

In my simple test, the size of the "bloomz-7b1" model file is 14.14G(9.98G+4.16G)
When I load it to the GPU and use the "tatsu-lab/alpaca" data to fine-tune (size: 24.2 MB) is about 38GB RAM (FP16, A100 40G GPU is OOM)

@wakeb98
Copy link

wakeb98 commented Jul 10, 2023

In my simple test, the size of the "bloomz-7b1" model file is 14.14G(9.98G+4.16G) When I load it to the GPU and use the "tatsu-lab/alpaca" data to fine-tune (size: 24.2 MB) is about 38GB RAM (FP16, A100 40G GPU is OOM)

did you solve the issue?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
3 participants
@tigerbaby1515 @roy-shih @wakeb98