numpy
matplotlib
pillow

[torch]
torch
