numpy>=1.19.0
pytorch=1.4.0
python=3.7.4
cuda=10.1