torch>=1.7.1
torchvision>=0.8.2
numpy
pandas
pynvml
tqdm
