diff --git a/docs/getting_started/clustering/clustering.md b/docs/getting_started/clustering/clustering.md index 306f98b0..8e990cff 100644 --- a/docs/getting_started/clustering/clustering.md +++ b/docs/getting_started/clustering/clustering.md @@ -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/). diff --git a/docs/getting_started/dim_reduction/dim_reduction.md b/docs/getting_started/dim_reduction/dim_reduction.md index 89110752..8422160f 100644 --- a/docs/getting_started/dim_reduction/dim_reduction.md +++ b/docs/getting_started/dim_reduction/dim_reduction.md @@ -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: diff --git a/docs/getting_started/tips_and_tricks/tips_and_tricks.md b/docs/getting_started/tips_and_tricks/tips_and_tricks.md index a6b684bf..70a7b7a4 100644 --- a/docs/getting_started/tips_and_tricks/tips_and_tricks.md +++ b/docs/getting_started/tips_and_tricks/tips_and_tricks.md @@ -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**