Python找不到火炬

Question

我想使用deepsaber。 为此，我必须安装 Torch。

我使用pip install torch torchvision 完成此操作。

运行cd scripts/generation和./script_generate.sh [path to song]后python告诉我：

Traceback (most recent call last):
  File "generate_stage1.py", line 16, in <module>
    from models import create_model
  File "/home/server/deepsaber/models/__init__.py", line 2, in <module>
    from .base_model import BaseModel
  File "/home/server/deepsaber/models/base_model.py", line 2, in <module>
    import torch
ModuleNotFoundError: No module named 'torch'
Traceback (most recent call last):
  File "generate_stage2.py", line 16, in <module>
    from models import create_model
  File "/home/server/deepsaber/models/__init__.py", line 2, in <module>
    from .base_model import BaseModel
  File "/home/server/deepsaber/models/base_model.py", line 2, in <module>
    import torch
ModuleNotFoundError: No module named 'torch'

使用import torch 就像使用python 时的魅力一样

对我能做什么有什么想法吗？

回购说：

pytorch（安装为torch或通过https://pytorch.org/get-started/locally/）

Answer 1

您很可能使用 Python 2 安装 torch，而 script_generate.sh 使用 Python 3（请参阅 here）：

# [...]
py=python3
# [...]

尝试运行pip3 install torch torchvision 或python3 -m pip install torch torchvision。另外，检查import torch 在使用python3 时是否有效。