deepspeedai / deepspeedai/DeepSpeedExamples

run_6.7b.sh On A6000 48G OOM

Open
#661 0 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

Dominant language
Python
Stars
6.8k
Forks
1.1k
Avg merge
2d 16h
Merged PRs (30d)
1

Description

This is an reproducticity issue !
I have run the script run_6.7b.sh,
**On 2 X A6000 48G ** -> OOM;
Have you test this script ?
I found you here indicate Single A6000 48G can run opt-6.7b, But i can't run it .

Contributor guide

No contributing guide indexed for this repository

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Start with applications/DeepSpeed-Chat/training/step1_supervised_finetuning/training_scripts/single_node/run_6.7b.sh and reproduce the reported run on two A6000 48G GPUs. Compare the result with the linked throughput and model-size claim, recording the exact OOM output and configuration. Done means identifying the cause and either correcting the script or documenting the verified hardware requirements.

Written by the indexing model from the issue text.

Assessment

Tech stack
bash, python
Domain
machine-learning
Issue type
Bug
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.