Факел установлен, но я не могу импортировать его в проект python компьютерного зрения в блокноте Jupyter

Я работаю над проектом компьютерного зрения, используя детектрон2. У меня возникли проблемы с установкой факела или его импортом в мой ноутбук Jupyter. Я работаю над Mac, на котором работает MACOS Catalina, версия Python3 в 3.8.2, и я использую Anaconda для своей среды разработки. На снимке экрана показано, что факел установлен, но при его импорте возникает ошибка введите здесь описание изображения


jupyter-notebook python pytorch torch detectron
person Muhammad Sameer    schedule 26.01.2021    source источник

Ответы (1)


arrow_upward
1
arrow_downward

Вы установили torch для python в /usr/local/lib/python3.9/site-packages

но, как вы сами сказали, вы

использование Anaconda для моей среды разработки

Итак, если ваш ноутбук Jupyter действительно настроен на использование вашей установки Python, поставляемой вместе с anaconda, вам нужно сделать

conda install pytorch torchvision torchaudio cudatoolkit=10.1 -c pytorch

Примечание:

  1. Убедитесь, какой интерпретатор Python на самом деле использует ваш jupyter, запустив

    import sys; print(sys.executable)

  2. Убедитесь, что вы устанавливаете torch для правильной среды conda (приведенная выше команда также укажет, какая именно)

  3. Установите версию cudatoolkit в соответствии с вашими потребностями. На официальном сайте также есть простой интерфейс для получения правильной команды.

person FlyingTeller    schedule 26.01.2021
comment
Спасибо, вы спасатель - person Muhammad Sameer; 26.01.2021