torch
torchvision
openstl @ git+https://github.com/YingtianDt/OpenSTL.git