ImportError：没有名为炬管的模块

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

我已经使用pip install torch torchvision

安装了pytorch

有人知道如何解决吗？

Answer 1

否则，您可以检查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

Answer 2

您尚未安装Torch。通过https://pytorch.org/get-started/locally/

用pip安装它

Answer 3

在Linux中，

which python
which pip

在Windows Power Shell中

where python
where pip

检查pip和python是否都来自同一父目录。

Answer 4

尝试运行python3中的所有内容：

python3 -m pip install torch torchvision

后跟：

python3 setup.py install

Answer 5

python 默认为 python2

所以你应该把 python setup.py install 改为 python3 setup.py install