导入错误：没有名为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

我已经使用pip install torch torchvision安装了pytorch。

有人知道如何解决这个问题吗？

- Rose Ben Ann

---

你尝试过使用 pip3 install torch torchvision 吗？我认为你可能在 Python 3 中运行设置脚本，但安装了 Python 2 的 pip 包。 - VietHTran

也许你有两个Python，你在其中一个Python中安装了torch，但你在另一个Python中运行代码。检查python -V和pip -V或使用python -m pip install ...。 - furas

我也遇到了同样的问题，我可以通过 pip list 和 pip3 list 看到它已经安装了。 - Digvijay Upadhyay

Answer 1

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

python3 -m pip install torch torchvision

接下来是:

python3 setup.py install

- Roshin Raphel

Answer 2

Python 默认使用的是 python2

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

- DucAnh

Answer 3

在Linux中，

which python
which pip

在Windows PowerShell中

where python
where pip

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

- Krishan Subudhi

Answer 4

您还没有安装Torch。请通过pip安装它：https://pytorch.org/get-started/locally/

- Minh-Long Luu

Answer 5

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

- lurad101