Closed 152334H closed 1 year ago
also deepspeed + mpi4py, for the deepspeed related examples
Thanks, I'll refine the setup.py. Is it now working if you manually install numpy? It is a little weird since you have torch installed but numpy is not there.
yeah I thought it would've came with torch too, but it did not (2.0.1+cu118).
it works fine with numpy (and others) manually installed, just wanted to make a public note for anyone else using it
Sure, you may have a same question about why torch doesn't install numpy dependency as well. Now the main
branch has numpy dependency added. Thanks for your information!
Thanks!
after following the instructions to build from source, i noticed that numpy is needed for the megablocks example:
but is not installed by setup.py because it is not in
install_requires
recommendation: add
numpy
there, also maybe addninja
as an optional for building (not sure how this could be done in practice)