PyTorch 已安装但未导入

Question

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

Answer 1

事实上，你应该做 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]])

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

PyTorch 已安装但未导入

PyTorch is installed but not imported

python-3.x

ubuntu-16.04

pytorch