0

我正在尝试开始使用TRT_pose，但是在尝试从 安装插件时遇到了困难setup.py，使用终端中的这一行：sudo python3 setup.py install --plugins。我得到错误

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

但是我已经安装了 torch、torchvision 和 torchaudio 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，所以我不知道为什么我仍然看到这个错误。我在虚拟机上运行 Ubuntu 20.4，并在其上安装了 Python 3.8.5。

4

1 回答 1

0

请接受我的建议：我还没有跑 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
  2. 如果您确实需要使用 sudo 一个选项是重新安装当前用户已经安装的依赖项，但是作为sudo
  3. 上述问题的一个潜在解决方法是创建从您的用户到sudo用户路径的符号链接，以便您的用户安装的相同库/二进制文件和 pip3 包也可用于sudo

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

于 2021-06-23T00:09:16.347 回答