Установка PyTorch совместно с CUDA

Смотреть материал на YouTube | RuTube

Это занятие мы с вами начнем с установки на локальный компьютер фреймворка PyTorch, который значительно облегчает построение и обучение самых разных нейронных сетей (НС). Но почему именно PyTorch? В действительности, на сегодняшний день, в этой области существуют два основных конкурирующих между собой фреймворка:

PyTorch и Tensorflow/Keras

Оба имеют довольно схожий функционал. Однако полагается, что использование PyTorch несколько проще для построения самых разнообразных архитектур НС, а также он обладает более удобным процессом отладки (debug) программного кода. По этим причинам PyTorch чаще используют в научных, исследовательских целях. В то время как Tensorflow/Keras больше зарекомендовал себя в коммерческих приложениях (в продакшене, production) благодаря, как правило, более высокой скорости обучения НС. Однако различия по скорости работы между этими двумя фреймворками не так существенны, а иногда и вовсе незаметны, чтобы отказываться от удобства PyTorch в пользу Tensorflow/Keras. Не зря по статистике PyTorch на сегодняшний день номер один в научных исследованиях. Мало того, изучив PyTorch, вы легко сможете перейти на Tensorflow/Keras и, наоборот.

Установка PyTorch

Фреймворком PyTorch можно воспользоваться в Google Colaboratory:

https://colab.google

Там все распространенные пакеты уже установлены, в том числе и PyTorch. Проверить это можно командами:

import torch
 
print(torch.__version__)

Должна отобразиться версия пакета PyTorch и возможность поддержки GPU – графического процессора.

Но, конечно, лучше и удобнее установить этот фреймворк на свой локальный компьютер. Подробная инструкция, как это делается, приведена на странице официальной документации:

https://pytorch.org/get-started/locally/

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

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124

если для возможности использования графического процессора фирмы Nvidia, либо командой:

pip3 install torch torchvision torchaudio

без использования GPU.

Видео по теме