--find-links https://download.pytorch.org/whl/cu113/torch_stable.html
torch==1.12.1+cu113 
torchaudio==0.12.1+cu113
torchvision==0.13.1
einops
colossalai==0.1.8