Skip to content

Инструкция по установке PyTorch

Ilya Zakharkin edited this page Jul 13, 2018 · 3 revisions

PyTorch - это фреймворк глубокого обучения, ключевой особенностью которого является построение графа вычислений "на лету", в отличие от тех же TensorFlow и Theano (и надстройки - Keras), в которых сначала нужно построить граф, а потом "скомпилировать" его и запустить. Он сравнительно новый, поэтому в нём нет ещё многих вещей, которые есть, например, в Caffe, однако он бурно развивается (он разрабатывается в стенах компании Facebook).

Вместе с PyTorch нужно установить torchvision - это набор готовых датасетов и моделей (они точно пригодятся Вам в процессе работы с PyTorch).

Сначала попробуйте установить PyTorch с главной страницы официального сайта.
Если всё же не получается, следующие команды скорее всего помогут:

ОС Что писать (последовательность действий) Где это писать
Linux conda install pytorch torchvision -c pytorch Terminal
MacOS conda install pytorch torchvision -c pytorch Terminal
Windows (64 bit) 1. conda install -c peterjc123 pytorch
2. pip install torchvision
или
2. pip install --no-deps torchvision
Командная строка