Skip to content

Commit

Permalink
Update installation instructions
Browse files Browse the repository at this point in the history
Signed-off-by: Ryan Wolf <[email protected]>
  • Loading branch information
ryantwolf committed May 16, 2024
1 parent e5c3b8c commit 82c0f4d
Showing 1 changed file with 24 additions and 4 deletions.
28 changes: 24 additions & 4 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -39,21 +39,41 @@ These modules are designed to be flexible and allow for reordering with few exce

NeMo Curator currently requires Python 3.10 and the GPU accelerated modules require CUDA 12 or above installed in order to be used.

### PyPi

NeMo Curator can be installed manually by cloning the repository and installing as follows -

For CPU only modules:
```
pip install .
pip install nemo-curator
```

For CPU + CUDA accelerated modules
For CPU + CUDA accelerated modules:
```
pip install --extra-index-url https://pypi.nvidia.com ".[cuda12x]"
pip install --extra-index-url https://pypi.nvidia.com nemo-curator[cuda12x]
```

### NeMo Framework Container

NeMo Curator is available in the [NeMo Framework Container](https://catalog.ngc.nvidia.com/orgs/nvidia/containers/nemo). The NeMo Framework Container provides an end-to-end platform for development of custom generative AI models anywhere. The latest release of NeMo Curator comes preinstalled in the container.
The latest release of NeMo Curator is preinstalled in the [NeMo Framework Container](https://catalog.ngc.nvidia.com/orgs/nvidia/containers/nemo). The NeMo Framework Container provides an end-to-end platform for development of custom generative AI models anywhere. If you want the latest commit inside the container, uninstall the existing version using
```
pip uninstall nemo-curator
```
And follow the instructions for installing from source below.

### Source

If you want to install the latest commit, please clone the repository and install with either of the follwing commands

For CPU only modules:
```
pip install .
```

For CPU + CUDA accelerated modules:
```
pip install --extra-index-url https://pypi.nvidia.com ".[cuda12x]"
```

## Usage

Expand Down

0 comments on commit 82c0f4d

Please sign in to comment.