No module named torch

1.从github上下载了ganomaly，准备看下效果，但在运行run_mnist.sh 命令时报错，具体报错信息如下：

2.运行.sh文件报错之后验证了pytorch,torch,torchvision，没有问题。

3.之后也单独重新安装了torch，但是还是报同样的错误。

用pip install安装下pytorch

系统默认是Python2吧