已安装Pytorch，但在Ubuntu 18.04上不起作用

Question

我正在尝试通过ubuntu 18.04上的pip安装Pytorch。我有python 3.6，我的笔记本电脑是HP-Pavilion笔记本电脑15

安装似乎是正确的，因为我收到消息：

  

成功安装收集到的软件包：火炬，火炬视觉   已安装Torch-1.3.1 + cpu torchvision-0.4.2 + cpu

我运行验证码，没关系

  from __future__ import print_function

  import torch

  x = torch.rand(5, 3)

  print(x)

但是，当我关闭终端或重新启动并尝试运行他的相同代码时，出现错误：

  

回溯（最近通话最近一次）：

     

文件“ torch.py​​”，第2行，在       进口火炬

     

AttributeError：模块'torch'没有属性'rand'

Answer 1

您如何执行python脚本？您正在使用哪个python？也许您以其他python版本安装了该软件包？

尝试为要使用的python设置别名：

alias python=/usr/local/bin/python3.6

然后pip使用您将一直使用的python别名安装软件包。

python pip install <package name>

Python现在将使用别名 python将软件包安装在python文件中-转到文件：/usr/local/bin/python3.6

让我知道错误是否仍然存在！

Answer 2

通过以下命令使用pip安装pytorch：

pip3 install torch==1.3.1+cpu torchvision==0.4.2+cpu -f https://download.pytorch.org/whl/torch_stable.html

有关任何参考，请通过pytorch的{​​{3}}。

Answer 3

将文件.py更改为另一个名称，在导入火炬时将其命名为torch.py​​，它将调用ur torch.py​​