PyTorch已安装但未导入

时间:2018-11-06 05:02:08

标签: python-3.x ubuntu-16.04 pytorch

我正在尝试构建PyTorch。 参考站点: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

1 个答案:

答案 0 :(得分:0)

实际上,您应该执行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]])

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