Как установить pytorch на ubuntu?

Пользователь

от amaya_bahringer , в категории: Python , год назад

Как установить pytorch на ubuntu?

Facebook Vk Ok Twitter LinkedIn Telegram Whatsapp

2 ответа

Пользователь

от enid , год назад

@amaya_bahringer 

Для установки PyTorch на Ubuntu можно использовать pip, командную строку следующей командой:

1
pip3 install torch torchvision


Если вы хотите установить конкретную версию PyTorch, вы можете использовать соответствующую команду, например:

1
pip3 install torch==1.7 torchvision==0.8


Если вам нужна cuda поддержка, нужно указать соответствующий терминальный аргумент:

1
pip3 install torch torchvision torchvision torchvision -f https://download.pytorch.org/whl/cu111/torch_stable.html


Обратите внимание, что необходимо совместимое с cuda версия для вашей версии cuda.

Пользователь

от forest , 6 месяцев назад

@amaya_bahringer 

Если вы предпочитаете установку PyTorch из исходного кода, следуйте инструкциям, представленным на официальном сайте PyTorch: https://pytorch.org/get-started/locally/


Кроме того, нужно установить дополнительные библиотеки, такие как numpy и matplotlib, которые могут быть полезны при работе с PyTorch:


1


pip3 install numpy matplotlib


После установки PyTorch вы можете проверить его работу в Python-оболочке. Создайте новый файл Python и импортируйте библиотеку PyTorch:


1


import torch


Если импорт прошел успешно и ошибок не возникло, значит, PyTorch успешно установлен на вашей системе.