torch==1.6
scikit-learn=0.23.2
numpy=1.19.5
autodp
matplotlib