Skip to content

[BUG] DeepSpeed on pypi not compatible with latest numpy #5671

@younesbelkada

Description

@younesbelkada

Describe the bug

Importing deepspeed on a python env with numpy>=2.0.0 fails:

  File "/miniconda3/envs/py39/lib/python3.9/site-packages/deepspeed/autotuning/scheduler.py", line 8, in <module>
    from numpy import BUFSIZE
E   cannot import name 'BUFSIZE' from 'numpy' (/miniconda3/envs/py39/lib/python3.9/site-packages/numpy/__init__.py)

To Reproduce

pip install deepspeed on a env with python>=3.9 and import deepspeed

Metadata

Metadata

Assignees

Labels

Type

No type
No fields configured for issues without a type.

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions