requirement for torch

[shawnzhangyx]

Hi Jingtian,

It looks like the module torch is also required, but not checked during installation.