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 ( устанавливается как горелка или через https://pytorch.org/get-started/locally/)

Answer 1

Скорее всего, вы установили torch, используя Python 2, а script_generate.sh использует Python 3 (см. здесь ):

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

Попробуйте запустить pip3 install torch torchvision или python3 -m pip install torch torchvision. Также проверьте, работает ли import torch при использовании python3.