首页 > 解决方案 > Pytorch 已安装但无法在 ubuntu 18.04 上运行

问题描述

我正在尝试在 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”

标签: pythonpytorchubuntu-18.04torchtorchvision

解决方案


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

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


推荐阅读