Установка пакетов PyTorch, torchvision, torchaudio, torchtext и torchcsprng для aarch64 / ARMv8 / ARM64

Пакеты PyTorch (включая torchvision, torchaudio, torchtext и torchcsprng) доступны в виде файлов колёс (whl), подходящих для устройств на базе aarch64 / ARMv8 / ARM64.

Установка с помощью conda

conda install -c kumatea numpy pytorch

На данный момент доступна поддержка torchvision.

Установка через pip

pip install torch -f https://cf.torch.kmtea.eu/whl/stable-cn.html

В качестве альтернативы можно использовать ссылку: pip install torch -f https://torch.kmtea.eu/whl/stable-cn.html. При необходимости можно добавить пакеты torchvision, torchaudio, torchtext и torchcsprng.

Обратите внимание, что эта команда устанавливает последнюю версию. Для выбора определённой версии обратитесь к разделу «Соответствующие версии».

Если вам нужно выбрать файл whl вручную, обратитесь к разделу release.

Использование Docker

docker run -it kumatea/pytorch

Чтобы загрузить образ, выполните команду docker pull kumatea/pytorch.

Для просмотра всех версий перейдите по ссылке.

FastAI

FastAI — это открытый высокоуровневый фреймворк глубокого обучения, основанный на PyTorch.

Установка с помощью conda

Рекомендуется использовать следующую команду:

conda install -c fastai -c kumatea fastai

Также можно установить fastbook с помощью команды:

conda install -c fastai -c kumatea fastbook

Установка через pip

Выполните команду:

pip install fastai -f https://cf.torch.kmtea.eu/whl/stable.html

Пакеты torch и torchvision будут установлены автоматически как зависимости.

Дополнительная информация

Поддержка CUDA / CUDNN

Поскольку в среде компиляции нет Nvidia GPU, поддержка cuda недоступна. Если вам нужна эта функция, используйте плату разработки Nvidia Jetson.

Среда компиляции

SoC: BCM2711 (четыре ядра A53). Архитектура: ARMv8 / ARM64 / aarch64. Операционная система: Debian Buster. Виртуализация: Docker. Щитс.ио/бейдж/питон-3.6 | 3.7 | 3.8 | 3.9 | 3.10 — синий

АРМ-в7 | в8 — синий

гитхаб.ком/КумаТи/питорч-арм

гитхаб. ком/ КумаТи / питорч-аарч64 / релизс / таг / в1.8.0

ффмпег точка орг

гитхаб точка ком/КумаТи/питорч-аач64/релизс/таг/в1.8.1

даунлоад точка питорч точка орг/вхл/торч_стабиле точка хтмл

серклси точка ком/ джи эйч / КумаТи / питорч-аарч64 точка сваг? стайл = сваг

серклси точка ком / джи эйч / КумаТи / питорч-аарч64

гитхаб точка ком/ КумаТи / питорч-аарч64 / релизс/ таг/ в1.9.0

гитхаб точка ком/ КумаТи / питорч-аарч64 / иссюс / 8

гитхаб точка ком/ конда фордж / минифордж

гитхаб точка ком/ КумаТи / тензорфлоу-аарч64

гитхаб точка ком/ КумаТи / питорч-аарч64 / релизс/ таг/ в1.10.0

гитхаб точка ком/ КумаТи / экст-вхл

анаконда точка орг / кума ти / питорч