ImportError: нет модуля с именем torch

Question

Я пытаюсь установить этот фреймворк с Github под названием torch2trt (здесь: https://github.com/NVIDIA-AI-IOT/torch2trt).

Я запустил следующий код в терминале Python, чтобы установить его :

git clone https://github.com/NVIDIA-AI-IOT/torch2trt
cd torch2trt
python setup.py install

, но в итоге появилось следующее сообщение об ошибке:

Traceback (most recent call last):
  File "setup.py", line 2, in <module>
    import torch
ImportError: No module named torch

Я уже установил pytorch, используя pip install torch torchvision

Кто-нибудь знает, как решить эту проблему ?

Answer 1

Попробуйте запустить все в python3:

python3 -m pip install torch torchvision

, а затем:

python3 setup.py install

Answer 2

Вы не установили Torch. Установите его с помощью pip через: https://pytorch.org/get-started/locally/

Answer 3

In Linux,

which python
which pip

In Windows power shell

where python
where pip

Проверить, находятся ли оба пункта и python из одного родительского каталога.

Answer 4

В противном случае вы можете проверить страницу PyTorch и попробовать другую версию. https://pytorch.org/get-started/locally/

Я бы сначала попробовал это.

pip install torch == 1.6.0 + cu101 torchvision == 0.7.0 + cu101 -f https://download.pytorch.org/whl/torch_stable.html