Skip to content
Open
Changes from all commits
Commits
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
12 changes: 9 additions & 3 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -397,18 +397,24 @@ uv sync --extra cu13
uv run python -c "import physicsnemo; print('PhysicsNeMo version:', physicsnemo.__version__)"
```

To install with optional feature extras (e.g., `nn-extras`):
To install with optional feature extras (e.g., `model-extras`):

```Bash
uv sync --extra cu13 --extra nn-extras
uv sync --extra cu13 --extra model-extras
```

For a CUDA 12 environment, replace `cu13` with `cu12`:

```Bash
uv sync --extra cu12 --extra nn-extras
uv sync --extra cu12 --extra model-extras
```

**Note:** Extras such as `gnns`, `nn-extras`, and `transformer-engine`
pull in packages that either ship only source distributions or require
wheels built for a specific PyTorch and CUDA version. These should be
installed with `--no-build-isolation`,
e.g. `uv sync --extra cu13 --extra nn-extras --no-build-isolation`.

### NVCR Container

The recommended PhysicsNeMo Docker image can be pulled from the
Expand Down