torch
opencv-python
psutil
torchvision
seaborn