Skip to content

[BUG] pip install graphistry[umap_learn] finds cupy import error when running umap even if engine=umap_learn #581

@silkspace

Description

@silkspace

Describe the bug
pip install graphistry[umap_learn] finds cupy import error when running umap even if engine=umap_learn. Pip installing cupy doesn't resolve problem.

To Reproduce

g.umap(engine='umap_learn')

Expected behavior
Confirmed that in google.collab this works, but it did not when client tried.

Actual behavior
Error:

    from cupy import _core  # NOQA
  File “/usr/local/lib/python3.10/dist-packages/cupy/_core/__init__.py”, line 3, in <module>
    from cupy._core import core  # NOQA
ImportError: libcuda.so.1: cannot open shared object file: No such file or directory
The above exception was the direct cause of the following exception:
Traceback (most recent call last):
  File “/usr/local/lib/python3.10/dist-packages/graphistry/utils/lazy_import.py”, line 11, in lazy_cudf_import
    import cudf  # type: ignore
  File “/usr/local/lib/python3.10/dist-packages/cudf/__init__.py”, line 12, in <module>
    import cupy
  File “/usr/local/lib/python3.10/dist-packages/cupy/__init__.py”, line 19, in <module>
    raise ImportError(f’’'
ImportError:
================================================================
Failed to import CuPy.

Browser environment (please complete the following information):

  • OS: [Windows]
  • Browser [chrome]
  • version []

Graphistry GPU server environment

  • Where run [Google Collab]

PyGraphistry API client environment

  • Where run [Collab Graphistry 2.35.9 Jupyter]
  • Version [0.34.3]
  • Python Version [e.g. Python 3.7.7]

Metadata

Metadata

Assignees

Labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions