torch
torchvision
opencv-python-headless
