--index-url https://download.pytorch.org/whl/cpu
numpy
scikit-learn
tqdm
torch
