我正在尝试构建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
答案 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]])
编辑:该版本对我来说完全没有问题。但是,如果您坚持要执行单元测试,也许其他人可以解决您的问题。