Skip to content

Conversation

etowahadams
Copy link
Contributor

The version of deepspeed specified in environment.yml will lead to a module 'deepspeed' has no attribute 'comm' error, so we need to upgrade it. The latest numpy version will lead to Could not infer dtype of numpy.int32 error, so I downgraded it. This lets all of the tests pass

- pytorch-lightning
- biopython
- numpy
- numpy==1.26.4
Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Currently, numpy is unversioned, causing numpy 2.0+ to be installed which leads to failing tests

- pytorch::pytorch-cuda=12.1
- pip:
- deepspeed==0.12.4
- deepspeed==0.14.5
Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Old version causes a module 'deepspeed' has no attribute 'comm' error

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant