sarman/dgx-spark-playbooks
1
0
Fork 0
mirror of https://github.com/NVIDIA/dgx-spark-playbooks.git synced 2026-04-25 11:23:52 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

chore: Regenerate all playbooks

Browse Source
This commit is contained in:
GitLab CI 2025-10-07 13:16:58 +00:00
parent 52be9adacc
commit ec9ace0a1a
1 changed files with 28 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
nvidia/pytorch-fine-tune/README.md
View File

@ -40,13 +40,32 @@ ALl files required for finetuning are included.
## Instructions ## Instructions
## Step 1. Pull the latest Pytorch container ## Step 1. Configure Docker permissions
To easily manage containers without sudo, you must be in the `docker` group. If you choose to skip this step, you will need to run Docker commands with sudo.
Open a new terminal and test Docker access. In the terminal, run:
```bash
docker ps
```
If you see a permission denied error (something like `permission denied while trying to connect to the Docker daemon socket`), add your user to the docker group:
```bash
sudo usermod -aG docker $USER
```
> **Warning**: After running usermod, you must log out and log back in to start a new
> session with updated group permissions.
## Step 2. Pull the latest Pytorch container
```bash ```bash
docker pull nvcr.io/nvidia/pytorch:25.09-py3 docker pull nvcr.io/nvidia/pytorch:25.09-py3
``` ```
## Step 2. Launch Docker ## Step 3. Launch Docker
```bash ```bash
docker run --gpus all -it --rm --ipc=host \ docker run --gpus all -it --rm --ipc=host \
@ -54,15 +73,18 @@ docker run --gpus all -it --rm --ipc=host \
-v ${PWD}:/workspace -w /workspace \ -v ${PWD}:/workspace -w /workspace \
nvcr.io/nvidia/pytorch:25.09-py3 nvcr.io/nvidia/pytorch:25.09-py3
-v $HOME/.cache/huggingface:/root/.cache/huggingface \
-v ${PWD}:/workspace -w /workspace \
nvcr.io/nvidia/pytorch:25.09-py3
``` ```
## Step 3. Install dependencies inside the contianer ## Step 4. Install dependencies inside the contianer
```bash ```bash
pip install transformers peft datasets "trl==0.19.1" "bitsandbytes==0.48" pip install transformers peft datasets "trl==0.19.1" "bitsandbytes==0.48"
``` ```
## Step 4: authenticate with huggingface ## Step 5: authenticate with huggingface
```bash ```bash
huggingface-cli login huggingface-cli login
@ -70,7 +92,7 @@ huggingface-cli login
##<Enter n for git credential> ##<Enter n for git credential>
``` ```
To run LoRA on Llama3 use the following command: To run LoRA on Llama3-8B use the following command:
```bash ```bash
python Llama3_8B_LoRA_finetuning.py python Llama3_8B_LoRA_finetuning.py
@ -80,6 +102,7 @@ To run qLoRA finetuning on llama3-70B use the following command:
```bash ```bash
python Llama3_70B_qLoRA_finetuning.py python Llama3_70B_qLoRA_finetuning.py
``` ```
To run full finetuning on llama3-3B use the following command: To run full finetuning on llama3-3B use the following command:
```bash ```bash
python Llama3_3B_full_finetuning.py python Llama3_3B_full_finetuning.py