scipy
tensorboard
einops
tqdm
pandas
pyyaml
torch==2.0.0
torchvision==0.15.1
torchaudio==2.0.1
-f https://download.pytorch.org/whl/cu118