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