Installer bug - wrong `apex` package installed

microsoft / torchscale

Foundation Architecture for (M)LLMs

https://aka.ms/GeneralAI

MIT License

3k stars 201 forks source link

Installer bug - wrong `apex` package installed #18

Closed jph00 closed 1 year ago

jph00 commented 1 year ago

If you pip install torchscale then via requirements it also installs apex from pypi. However, apex on pypi is not Nvidia's apex, but is an unrelated project with many deps. As a result, many other additional deps are also pulled into the installation.

Nvidia's apex is currently not pip-installable, and therefore it should not be listed as a requirement. cc @shumingma

shumingma commented 1 year ago

Thanks for catching that! I've uploaded a fix to torchscale==0.1.2 by removing apex from the installation dependency and making it an optional package.

jph00 commented 1 year ago

Many thanks. Closing this.