Skip to content

Commit 3b03c97

Browse files
dlwh-golemdlwh
andauthored
docs: align local GPU tutorial with uv (#3662)
## Summary - switch the local GPU tutorial example to `uv run python ...` to match the repos documented execution flow - fix the malformed fenced code block for `XLA_PYTHON_CLIENT_MEM_FRACTION` in the same tutorial section ## Validation - `uv sync --package marin --group docs` - `uv run mkdocs build --strict` Refs #12 Co-authored-by: dlwh-golem <david.hall@openathena.ai>
1 parent c4bdde4 commit 3b03c97

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

docs/tutorials/local-gpu.md

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -53,9 +53,10 @@ See [JAX's installation guide](https://jax.readthedocs.io/en/latest/installation
5353

5454
!!! tip
5555
If you are using a DGX Spark or similar machine with unified memory, you may need to dramatically reduce the memory that XLA preallocates for itself. You can do this by setting the `XLA_PYTHON_CLIENT_MEM_FRACTION` variable, to something like 0.5:
56-
`bash
57-
export XLA_PYTHON_CLIENT_MEM_FRACTION=0.5
58-
`
56+
57+
```bash
58+
export XLA_PYTHON_CLIENT_MEM_FRACTION=0.5
59+
```
5960

6061
You can also set this in your `.bashrc` or `.zshrc` file.
6162
```bash
@@ -72,7 +73,7 @@ Let's start by running the tiny model training script (GPU version) [`experiment
7273
```bash
7374
export MARIN_PREFIX=local_store
7475
export WANDB_ENTITY=...
75-
python3 experiments/tutorials/train_tiny_model_gpu.py --prefix local_store
76+
uv run python experiments/tutorials/train_tiny_model_gpu.py --prefix local_store
7677
```
7778

7879
The `prefix` is the directory where the output will be saved. It can be a local directory or anything fsspec supports,

0 commit comments

Comments
 (0)