torch>=1.10.0
torchvision>=0.11.2
numpy>=1.18.4
kaleido>=0.0.1
tqdm>=4.48.0
cupy-cuda102
matplotlib>=3.3.1
gdown>=4.0.2
h5py>=2.10.0
