Skip to content
Merged
Show file tree
Hide file tree
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
21 changes: 16 additions & 5 deletions docs/getting_started/clustering/clustering.md
Original file line number Diff line number Diff line change
Expand Up @@ -110,12 +110,23 @@ The great thing about using cuML's HDBSCAN implementation is that it supports ma
However, it is still possible to calculate the topic-document probability matrix for the data on which the model was trained (i.e., `.fit` and `.fit_transform`).

!!! note
If you want to install cuML together with BERTopic using Google Colab, you can run the following code:
To install cuML with BERTopic, run these commands:

**For CUDA 12:**
```bash
!pip install cuml-cu12
!pip install bertopic
!pip install cudf-cu11 dask-cudf-cu11 --extra-index-url=https://pypi.nvidia.com
!pip install cuml-cu11 --extra-index-url=https://pypi.nvidia.com
!pip install cugraph-cu11 --extra-index-url=https://pypi.nvidia.com
!pip install --upgrade cupy-cuda11x -f https://pip.cupy.dev/aarch64
```

**For CUDA 13:**
```bash
!pip install cuml-cu13
!pip install bertopic
```

!!! warning
Install cuML first, then BERTopic. Installing both in a single command can fail due to pip resolver limitations with CUDA runtime dependencies.

**Note:** cuML is already installed on Google Colab.

For more detailed information on installing cuML, including additional dependencies and platform-specific instructions, see the [RAPIDS installation guide](https://docs.rapids.ai/install/).
21 changes: 16 additions & 5 deletions docs/getting_started/dim_reduction/dim_reduction.md
Original file line number Diff line number Diff line change
Expand Up @@ -97,16 +97,27 @@ topic_model = BERTopic(umap_model=umap_model)
```

!!! note
If you want to install cuML together with BERTopic using Google Colab, you can run the following code:
To install cuML with BERTopic, run these commands:

**For CUDA 12:**
```bash
!pip install cuml-cu12
!pip install bertopic
!pip install cudf-cu11 dask-cudf-cu11 --extra-index-url=https://pypi.nvidia.com
!pip install cuml-cu11 --extra-index-url=https://pypi.nvidia.com
!pip install cugraph-cu11 --extra-index-url=https://pypi.nvidia.com
!pip install --upgrade cupy-cuda11x -f https://pip.cupy.dev/aarch64
```

**For CUDA 13:**
```bash
!pip install cuml-cu13
!pip install bertopic
```

!!! warning
Install cuML first, then BERTopic. Installing both in a single command can fail due to pip resolver limitations with CUDA runtime dependencies.

**Note:** cuML is already installed on Google Colab.

For more detailed information on installing cuML, including additional dependencies and platform-specific instructions, see the [RAPIDS installation guide](https://docs.rapids.ai/install/).


## **Skip dimensionality reduction**
Although BERTopic applies dimensionality reduction as a default in its pipeline, this is a step that you might want to skip. We generate an "empty" model that simply returns the data pass it to:
Expand Down
21 changes: 16 additions & 5 deletions docs/getting_started/tips_and_tricks/tips_and_tricks.md
Original file line number Diff line number Diff line change
Expand Up @@ -181,16 +181,27 @@ embeddings = normalize(embeddings)
However, it is still possible to calculate the topic-document probability matrix for the data on which the model was trained (i.e., `.fit` and `.fit_transform`).

!!! note
If you want to install cuML together with BERTopic using Google Colab, you can run the following code:
To install cuML with BERTopic, run these commands:

**For CUDA 12:**
```bash
!pip install cuml-cu12
!pip install bertopic
!pip install cudf-cu11 dask-cudf-cu11 --extra-index-url=https://pypi.nvidia.com
!pip install cuml-cu11 --extra-index-url=https://pypi.nvidia.com
!pip install cugraph-cu11 --extra-index-url=https://pypi.nvidia.com
!pip install --upgrade cupy-cuda11x -f https://pip.cupy.dev/aarch64
```

**For CUDA 13:**
```bash
!pip install cuml-cu13
!pip install bertopic
```

!!! warning
Install cuML first, then BERTopic. Installing both in a single command can fail due to pip resolver limitations with CUDA runtime dependencies.

**Note:** cuML is already installed on Google Colab.

For more detailed information on installing cuML, including additional dependencies and platform-specific instructions, see the [RAPIDS installation guide](https://docs.rapids.ai/install/).


## **Lightweight installation**

Expand Down