ModuleNotFoundError:安装TRT\U pose时没有名为“torch”的模块

Question

我正在尝试开始使用TRT_pose，但在尝试从setup.py安装插件时遇到了问题，在terminal: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，所以我不知道为什么我仍然看到这个错误。我在虚拟机上运行Ubuntu20.4，并安装了Python 3.8.5

Answer 1

请接受我的建议：我还没有跑过trt_姿势

听起来这些问题与默认用户和超级用户的安装有关

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

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

1. 您需要使用sudo吗？如果您已经为您的用户安装了依赖项（PyTorch、CUDA等），那么您可以按照安装指南进行操作，而不必在命令前面加上sudo
2. 如果您确实需要使用sudo，一个选项是重新安装当前用户已经安装的依赖项，但是作为sudo
3. 对于上述问题，一个潜在的黑客解决方法是创建从用户到sudo用户路径的符号链接，以便用户安装的相同库/二进制文件和pip3包也可用于sudo

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