numpy
matplotlib
opencv

[cuda]
torch
