numpy
torch
torchvision
Pillow
nvidia-ml-py
piq
pot
scikit-learn
