deepspeedai / deepspeedai/DeepSpeedExamples
run step3 66b opt report an error out of memory
Open
@mrwyattii is already working on this.
Since Oct 10, 2023.
- Dominant language
- Python
- Stars
- 6.8k
- Forks
- 1.1k
- Avg merge
- 2d 16h
- Merged PRs (30d)
- 1
Description
When I run step three I get an out of memory error. the program error location is 1072 line in /usr/local/lib/python3.10/dist-packages/deepspeed-0.10.1+cbf2f61a-py3.10.egg/deepspeed/runtime/engine.py （if not self.dont_change_device:
self.module.to(self.device)）
The program completely copied the model onto the gpu causing a memory overflow.What's the solution
I used the A100 machine and used offload and offload_reference_model when I resubmitted the program.look forward to your reply
Contributor guide
No contributing guide indexed for this repository
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Assessment
This issue has not been assessed yet.