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

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

от amaya_bahringer , в категории: Python , 5 месяцев назад

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

Facebook Vk Ok Twitter LinkedIn Telegram Whatsapp

1 ответ

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

от enid , 5 месяцев назад

@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.