ModuleNotFoundError:安装TRT_pose时没有名为“torch”的模块

Question

我正在尝试开始使用TRT_pose，但在终端中使用以下代码行从setup.py安装插件时遇到了问题：sudo python3 setup.py install --plugins。我得到了错误

  File "setup.py", line 2, in <module>
    import torch
ModuleNotFoundError: No module named 'torch'

但是我已经在pip3 install torch==1.9.0+cpu torchvision==0.10.0+cpu torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html上安装了torch，torchvision和torchaudio，所以我不知道为什么我仍然看到这个错误。我在虚拟机上运行Ubuntu 20.4，并且安装了Python 3.8.5。

Answer 1

请对我的建议持保留态度:我还没有运行过trt_pose。

听起来问题似乎与作为默认用户与超级用户安装的内容有关。

python3 -c "import torch;print(torch.__version__)"输出1.9.0+cpu，但sudo python3 setup.py install --plugins在导入pytorch时崩溃的事实表明，PyTorch是为您的用户安装的，而不是为超级用户安装的。

您有多个选项，其中一些比其他选项更简单：

1. 你需要使用sudo吗？如果您已经有依赖项(PyTorch、CUDA等)为您的用户安装的sudo

，也许您可以按照安装指南操作，而无需在命令前添加sudo

• If您确实需要使用sudo一个选项是重新安装您的当前用户已经安装的依赖项，但是由于sudo

• A可能比较麻烦，上面的变通方法是从您的用户建立指向sudo用户路径的符号链接，以便您的用户安装的相同的库/二进制文件和pip3包也可用于sudo

由于trt_pose是在nVidia硬件上运行的，因此值得仔细检查它是你需要使用的PyTorch的cpu版本，还是gpu + cuda版本。