Skip to content

[BUG] Direct use of numba.cuda.gpus lead to error that module numba has no attribute cuda #2198

@lezhangamd

Description

@lezhangamd

Description

After installing recommenders[gpu], doing len(numba.cuda.gpus) will lead to the following error

>>> len(numba.cuda.gpus)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
AttributeError: module 'numba' has no attribute 'cuda'

Changing to from numba import cuda and then cuda.gpus will resolve the issue.

In which platform does it happen?

Linux Ubuntu 22.04

How do we replicate the issue?

See above.

Expected behavior (i.e. solution)

See above.

Willingness to contribute

  • Yes, I can contribute for this issue independently.
  • Yes, I can contribute for this issue with guidance from Recommenders community.
  • No, I cannot contribute at this time.

Other Comments

Metadata

Metadata

Assignees

No one assigned

    Labels

    bugSomething isn't working

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions