databricks / megablocks

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How to pip install the latest megablocks?

fanshiqing opened this issue · comments

Hi guys, how to install the lasted megablocks version?
It seems that the latest 0.3.3 just cannot be found by pip.

pip install megablocks==0.3.3
Defaulting to user installation because normal site-packages is not writeable
Looking in indexes: https://pypi.org/simple, https://pypi.ngc.nvidia.com
ERROR: Could not find a version that satisfies the requirement megablocks==0.3.3 (from versions: 0.0.1, 0.0.2)
ERROR: No matching distribution found for megablocks==0.3.3

From https://pypi.org/project/megablocks/, it seems currently only 0.0.2 is supported...

Currently I build megeblocks from local source as an workaround.
pip install -e .

Hello! It's a bit of a long story, but we can't put our latest packages on PyPi because they rely on Triton nightly and PyPi doesn't allow external dependencies. I recommend installing MegaBlocks via the git repo - for example, to install main:

pip install git+https://github.com/stanford-futuredata/megablocks.git@main

FYI, our PyPi packages are now up to date :)