torch>=1.7.0
numpy
unittest
