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

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

python pip install <package name>

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

让我知道错误是否仍然出现！

Answer 2

0
投票

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