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

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

@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 успешно установлен на вашей системе.