#python #pytorch
Вопрос:
Я хочу установить PyTorch 0.4.0 с cuda9.0 для использования этого репозитория.
Однако после установки, когда я пытаюсь импортировать PyTorch import torch
, я получаю следующую ошибку:
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
ModuleNotFoundError: No module named 'torch'
Я слежу за официальным сайтом pytorch, чтобы установить пакет.
Вот команды, которые я использовал для установки pytorch:
conda install pytorch=0.4.0 cuda90 -c pytorch
Комментарии:
1. Если conda успешно установила его, то вы не используете тот же интерпретатор python, что и conda.
2. Я думаю, что да, потому что я установил его в conda venv и использую python из самого этого venv.
3.
which python
Сначала проверьте это.