MaartenGr / BERTopic

Leveraging BERT and c-TF-IDF to create easily interpretable topics.
https://maartengr.github.io/BERTopic/
MIT License
6.2k stars 765 forks source link

Installing/Importing cuml HDBSCAN and UMAP issue #1917

Open jwratten opened 7 months ago

jwratten commented 7 months ago

Hi,

I used the following code to install cuml.

!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 cupy-cuda11x -f https://pip.cupy.dev/aarch64

However, when importing cuml.cluster and cuml.manifold I get the following error message. I am using databricks. Any help would be appreciated.

image

Thanks,

James

MaartenGr commented 7 months ago

Could you perhaps post this issue on the cuML repo? I think they will be much better in helping you out with this issue since this relates to cuML.

beckernick commented 7 months ago

This error generally appears when you're using an environment with incompatible versions of cuDF and Apache Arrow. This issue may provide clarity, but if not please file a fresh issue on the cuML or cuDF repository.