torch
torchvision
torchaudio
numpy
matplotlib
scipy
opencv-python
tensorboard
h5py
tqdm
einops