Это занятие мы с
вами начнем с установки на локальный компьютер фреймворка 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.