mirror of
https://github.com/NVIDIA/dgx-spark-playbooks.git
synced 2026-04-22 01:53:53 +00:00
chore: Regenerate all playbooks
This commit is contained in:
parent
e4ca6905af
commit
a1c0772ba6
@ -103,7 +103,7 @@ git clone https://github.com/NVIDIA/dgx-spark-playbooks
|
||||
> This command will download a base image and build a container locally to support this environment.
|
||||
|
||||
```bash
|
||||
cd jax/assets
|
||||
cd dgx-spark-playbooks/nvidia/jax/assets
|
||||
docker build -t jax-on-spark .
|
||||
```
|
||||
|
||||
|
||||
@ -93,7 +93,7 @@ huggingface-cli login
|
||||
|
||||
```bash
|
||||
git clone https://github.com/NVIDIA/dgx-spark-playbooks
|
||||
cd nvidia/pytorch-fine-tune/assets
|
||||
cd dgx-spark-playbooks/nvidia/pytorch-fine-tune/assets
|
||||
```
|
||||
|
||||
## Step7: Run the fine-tuning recipes
|
||||
|
||||
@ -87,6 +87,7 @@ docker run --gpus all --ulimit memlock=-1 -it --ulimit stack=67108864 --entrypoi
|
||||
```bash
|
||||
pip install transformers peft datasets "trl==0.19.1"
|
||||
pip install --no-deps unsloth unsloth_zoo
|
||||
pip install hf_transfer
|
||||
```
|
||||
|
||||
## Step 5. Build and install bitsandbytes inside Docker
|
||||
@ -99,7 +100,7 @@ pip install --no-deps bitsandbytes
|
||||
Curl the test script [here](https://github.com/NVIDIA/dgx-spark-playbooks/blob/main/nvidia/unsloth/assets/test_unsloth.py) into the container.
|
||||
|
||||
```bash
|
||||
curl -O https://github.com/NVIDIA/dgx-spark-playbooks/blob/main/nvidia/unsloth/assets/test_unsloth.py
|
||||
curl -O https://raw.githubusercontent.com/NVIDIA/dgx-spark-playbooks/refs/heads/main/nvidia/unsloth/assets/test_unsloth.py
|
||||
```
|
||||
|
||||
We will use this test script to validate the installation with a simple fine-tuning task.
|
||||
|
||||
Loading…
Reference in New Issue
Block a user