python - Pytorch 已安装但无法在 ubuntu 18.04 上运行

Question

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

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

安装采集包：torch、torchvision 成功安装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：模块“火炬”没有属性“rand”

score 1 · Accepted Answer

你是如何执行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

让我知道错误是否仍然发生！

score 0 · Accepted Answer

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

score 0 · Accepted Answer

通过以下命令使用 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 回答 3