Как установить 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.