PyTorch установлен, но не импортирован - PullRequest
Новая
       39

PyTorch установлен, но не импортирован

0 голосов
/ 06 ноября 2018

Я пытаюсь собрать 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 голосов
/ 26 декабря 2018

На самом деле, вы должны сделать 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]])

Edit: версия работает без проблем для меня. Но если вы настаиваете на выполнении модульного теста, возможно, другие люди могут решить вашу проблему.

...