mirror of
https://github.com/NVIDIA/dgx-spark-playbooks.git
synced 2026-04-25 11:23:52 +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.
|
> This command will download a base image and build a container locally to support this environment.
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
cd jax/assets
|
cd dgx-spark-playbooks/nvidia/jax/assets
|
||||||
docker build -t jax-on-spark .
|
docker build -t jax-on-spark .
|
||||||
```
|
```
|
||||||
|
|
||||||
|
|||||||
@ -93,7 +93,7 @@ huggingface-cli login
|
|||||||
|
|
||||||
```bash
|
```bash
|
||||||
git clone https://github.com/NVIDIA/dgx-spark-playbooks
|
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
|
## Step7: Run the fine-tuning recipes
|
||||||
|
|||||||
@ -87,6 +87,7 @@ docker run --gpus all --ulimit memlock=-1 -it --ulimit stack=67108864 --entrypoi
|
|||||||
```bash
|
```bash
|
||||||
pip install transformers peft datasets "trl==0.19.1"
|
pip install transformers peft datasets "trl==0.19.1"
|
||||||
pip install --no-deps unsloth unsloth_zoo
|
pip install --no-deps unsloth unsloth_zoo
|
||||||
|
pip install hf_transfer
|
||||||
```
|
```
|
||||||
|
|
||||||
## Step 5. Build and install bitsandbytes inside Docker
|
## 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.
|
Curl the test script [here](https://github.com/NVIDIA/dgx-spark-playbooks/blob/main/nvidia/unsloth/assets/test_unsloth.py) into the container.
|
||||||
|
|
||||||
```bash
|
```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.
|
We will use this test script to validate the installation with a simple fine-tuning task.
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user