Skip to content

Updated docker build command with the correct context #888

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

Open
wants to merge 2 commits into
base: main
Choose a base branch
from
Open

Updated docker build command with the correct context #888

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
e5db782
Updated docker build command with the correct context
aravneelaws Nov 4, 2025
004da41
Removed incomplete line - issue #813
aravneelaws Nov 4, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 1 addition & 3 deletions 3.test_cases/pytorch/FSDP/slurm/README.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ You will first build the container image with the command below:


```bash
docker build -f ../Dockerfile -t fsdp:pytorch2.7.1 .
docker build -f ../Dockerfile -t fsdp:pytorch2.7.1 ../.
```

You will then convert the container image to a squash file via Enroot:
Expand Down Expand Up @@ -71,8 +71,6 @@ Also, under `User Variables` make sure to adjust `GPUS_PER_NODE` to match the nu

You can also adjust the training parameters in `TRAINING_ARGS` (for example, to increase batch size). Additional parameters can be found in `model/arguments.py`. Note that we use the same directory for both `--checkpoint_dir` and `--resume_from_checkpoint`. If there are multiple checkpoints, `--resume_from_checkpoint` will automatically select the most recent one. This way if our training is interupted for any reason, it will automatically pick up the most recent checkpoint.

If you are using a container image, you need to uncomment the line below in the

### Llama 3.1 8B training

To launch your training for Llama 3.1 8B, run
Expand Down
Loading