[BUG] CI: ModuleNotFoundError: No module named 'cupy' #1431
Closed
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.