--extra-index-url https://download.pytorch.org/whl/cu116
torch==1.12.1
ninja 
triton==2.0.0.dev20221005
deepspeed==0.7.5
#deepspeed-mii==0.0.3