Skip to content

[BUG] CI: ModuleNotFoundError: No module named 'cupy' #1431

Closed
@fumihwh

Description

Describe the bug
PR's check failed due to ModuleNotFoundError: No module named 'cupy' url

WARNING: deepspeed 0.5.4+a415a0e does not provide the extra '1bit' shows up in Install DeepSpeed step. Related code is .github/workflows/main.yml:L61, pip install .[dev,1bit].

In setup.py:L52, extras_require contains three options, 1bit_adam, readthedocs and dev. And 1bit_adam will install cupy in L62:extras_require['1bit_adam'].append(cupy)

So the suggest fix is change 1bit_adam to 1bit or vice versa.

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