torch  # pytorch cudatoolkit
torchvision
opencv-contrib-python # opencv
