numpy
matplotlib
opencv-python

[cuda]
torch
