Pytorch已安装，但未在ubuntu 18.04上工作

Question

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

安装似乎是正确的，因为我收到了这样的信息：

安装收集的软件包:火炬，火炬视觉成功安装火炬-1.3.1+cpu火炬视觉-0.4.2+cpu

我运行验证代码，这样就可以了。

  from __future__ import print_function

  import torch

  x = torch.rand(5, 3)

  print(x)

但是，当我关闭终端或重新启动并尝试运行相同的代码时，会得到以下错误：

回溯(最近一次调用)： 文件"torch.py"，第2行，在导入火炬中 AttributeError:模块“火炬”没有属性'rand‘

Answer 1

如何执行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安装py手电筒：

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

要想得到任何参考，请通过“火把的官方网站”。

Answer 3

将文件.py更改为另一个名称，您在导入torch时命名为torch.py，它将调用您的torch.py