时间戳

首页 > 解决方案 > PyTorch 已安装但未导入

python-3.x - PyTorch 已安装但未导入

问题描述

我正在尝试构建 PyTorch。参考站点：<a href="https://github.com/hughperkins/pytorch" rel="nofollow noreferrer">https://github.com/hughperkins/pytorch

但是，当我们进行单元测试时，会出现以下错误。

ImportError while importing test module '/home/usr2/pytorch/test/testByteTensor.py'.
Hint: make sure your test modules/packages have valid Python names.
Traceback:
test/testByteTensor.py:2: in <module>
    import PyTorch
E   ImportError: No module named 'PyTorch'
__________________ ERROR collecting test/testDoubleTensor.py ___________________

我知道 PyTorch 没有被导入。但确认安装了pytorch。有没有办法解决这个问题？

环境

ubuntu 16.04
python3.5
cuda9.2

标签： python-3.xubuntu-16.04pytorch

解决方案

实际上，您应该这样做，import torch而不是import PyTorch
这对我有用：（我使用 conda 安装了它）

>>> import torch

>>> torch.version
>>> <module 'torch.version' from '/home/koke_cacao/miniconda3/envs/ml/lib/python3.6/site-packages/torch/version.py'>

>>> print(torch.__version__)
>>> 0.4.1.post2

>>> a = torch.FloatTensor(2,3)
>>> tensor([[-7.4368e-13,  3.0911e-41, -9.6122e-13],
    [ 3.0911e-41, -7.3734e-13,  3.0911e-41]])

编辑：该版本对我来说完全没有问题。但是如果你坚持执行单元测试，也许其他人可以解决你的问题。

推荐阅读