1 В избранное 0 Ответвления 0

OSCHINA-MIRROR/kumatea-pytorch-aarch64

Присоединиться к Gitlife
Откройте для себя и примите участие в публичных проектах с открытым исходным кодом с участием более 10 миллионов разработчиков. Приватные репозитории также полностью бесплатны :)
Присоединиться бесплатно
Клонировать/Скачать
Внести вклад в разработку кода
Синхронизировать код
Отмена
Подсказка: Поскольку Git не поддерживает пустые директории, создание директории приведёт к созданию пустого файла .keep.
Loading...
README.md

pytorch-aarch64

Deprecation

Since [v 1.8.0, PyTorch имеет официально предоставленные] колёса для aarch64. Этот проект выполнил свою миссию.


[![Build Status][50]][51] [![ver][59]][16] [![Coverage][43]][16]

Колёса и образы Docker для aarch64 / ARMv8 / ARM64 устройств:

  • PyTorch,
  • vision,
  • audio,
  • text,
  • csprng.

中文版 ([для Gitee][31]) | [GitHub][32] | [Web][33] | [TF][55]


Установка

Conda 🆕 (рекомендуется)

conda install -c kumatea pytorch

Возможно, потребуется установить numpy: conda install -c kumatea pytorch numpy

cpuonly в официальном руководстве по установке не требуется, но поддерживается: conda install -c kumatea pytorch numpy cpuonly

Pip

Не рекомендуется использовать pip для установки из этого источника. Вместо этого установите из официального индекса PyPI:

pip install torch

Установить отсюда...

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

При необходимости добавьте пакеты torchvision, torchaudio, torchtext, torchcsprng и другие.

Рассмотрите возможность использования готовых колёс для ускорения установки: pip install torch -f https://torch.kmtea.eu/whl/stable.html -f https://ext.kmtea.eu/whl/stable.html

(Для пользователей в Китае, пожалуйста, используйте CDN (README_zh.md#安装))

> Примечание: эта команда устанавливает последнюю версию. > Для выбора конкретной версии, пожалуйста, проверьте раздел «Пользовательские сборки».

Чтобы выбрать файлы whl вручную, пожалуйста, ознакомьтесь с [релизами][16].

Docker (не рекомендуется)

docker run -it kumatea/pytorch

Для извлечения образа выполните команду docker pull kumatea/pytorch.

Чтобы проверить все доступные теги, [нажмите здесь][34].


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://torch.kmtea.eu/whl/stable.html

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


Пользовательские сборки

torch torchvision torchaudio torchtext torchcsprng Статус python
master
nightly
master
nightly
master
nightly
master
nightly
master
nightly
>=3.6
1.10.0 0.11.1
0.11.0
0.10.0 0.11.0 [![passing][2]][56] >=3.6
1.9.1 0.10.1 0.9.1 0.10.1 >=3.6
1.9.0[i] 0.10.0 0.9.0 0.10.0 [![passing][2]][52] >=3.6
1.8.1 0.9.1[i] 0.8.1 0.9.1 0.2.1 [![passing][2]][48] >=3.6
1.8.0[i] 0.9.0 0.8.0 0.9.0 0.2.0 [![passing][2]][46] >=3.6
1.7.1 0.8.2 0.7.2 0.8.1 0.1.4 [![passing][2]][18] >=3.6
1.7.0 0.8.1
0.8.0
0.7.0 0.8.0 0.1.3 [![passing][2]][12] >=3.6
1.6.0[i] 0.7.0 0.6.0 0.7.0 0.1.2
0.1.1
0.1.0
[![passing][2]][10] >=3.6
1.5.1 0.6.1 0.5.1 0.6.0 [![passing][2]][35] >=3.5
1.5.0 0.6.0 0.5.0 0.6.0 [![passing][2]][36] >=3.5
1.4.1
1.4.0
0.5.0 0.4.0 0.5.0 [![passing][2]][37] ==2.7, >=3.5, <=3.8
1.3.1 0.4.2 ==2.7, >=3.5, <=3.7
1.3.0 0.4.1 ==2.7, >=3.5, <=3.7
1.2.0 0.4.0 ==2.7, >=3.5, <=3.7
1.1.0 0.3.0 ==2.7, >=3.5, <=3.7
<=1.0.1 0.2.2 ==2.7, >=3.5, <=3.7
  • Соответствующие версии torch и torchaudio
  • Соответствующие версии torch и torchtext

More Info

click to expand...

FAQ

  • Q: Работает ли это на Raspberry Pi? A: Да, если архитектура SoC — aarch64. Должен работать на всех чипах ARMv8.
  • Q: Поддерживает ли это CUDA / CUDNN? A: Нет. Подробнее см. здесь (#cuda--cudnn-support).
  • Q: Работает ли это на Nvidia Jetson? A: Да, но очень медленно. Каждая плата Nvidia Jetson содержит графический процессор Nvidia, но в этом проекте создаются только колёса для процессора. Чтобы лучше использовать оборудование, создайте его самостоятельно (build/torch.sh).

Отличие от официальных колёс

В большинстве случаев рекомендуется просто использовать официальные колёса, и они также будут установлены через pip по умолчанию, даже с -f. Колёса здесь скомпилированы из исходного кода на rpi 4b+, и предназначены для кодов, которые вылетали на официальных колёсах из-за использования некоторых неподдерживаемых инструкций. Используйте колёса torch здесь только в случае возникновения проблем, таких как #8 [53].

О Python 3.10

На момент фиксации этого изменения (v1.9.0) не было выпущено ни одной стабильной версии Python 3.10.0, Numpy 1.21.0 (который добавляет поддержку Python 3.10), или PyTorch 1.9.0 для Python 3.10. Если будет обнаружена какая-либо критическая проблема, я могу перестроить колесо после выпуска стабильных версий.

О PyTorch v1.8.0

  • Начиная с версии v1.8.0, наконец-то были выпущены официальные колёса PyTorch для aarch64!
    • Чтобы использовать официальные колёса, используйте эту ссылку на индекс: https://torch.kmtea.eu/whl/pfml.html, где pfml обозначает prefer-manylinux здесь. Колёса manylinux будут устанавливаться по умолчанию.
  • Колёса torchvision созданы с поддержкой FFmpeg [47]. Для колёс без неё установите torchvision==0.9.0+slim.

О PyTorch v1.6.0

При сборке PyTorch v1.6.0 была обнаружена фатальная ошибка, и этот патч был применён. Патч был позже объединён в основную ветку.

Об ошибке RuntimeError при импорте

Если вы видите что-то подобное при импорте torch:

RuntimeError: модуль скомпилирован против API версии 0xe, но эта версия numpy — 0xd

Обновите версию numpy: pip install -U numpy.

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

Поскольку среда сборки (как указано ниже) не содержит графического процессора Nvidia, колёса не могут быть созданы с поддержкой cuda. Если вам это нужно, пожалуйста, используйте плату Nvidia Jetson для запуска кода сборки (build/torch.sh).

Среда сборки

Хост: Raspberry Pi 4 Model B SoC: BCM2711 (четырёхъядерный A53) Архитектура: ARMv8 / ARM64 / aarch64 ОС: Debian Buster GCC: v8.3.0 Виртуализация: Docker

Производительность

Дата тестирования: 2021-10-29 Скрипт: bench.py (test/bench.py)

Чем меньше время выполнения, тем лучше

Платформа Характеристики Обучение Прогнозирование Версия
aarch64 BCM2711 (4x Cortex-A72) 1:48:44 11,506.080 мс 1.10.0 3.9.7
aarch64 QUALCOMM Snapdragon 845 Н/Д 4,821.148 мс (24x) 1.10.0 3.9.7
amd64 INTEL Core i5-6267U 162.964 с 140.680 мс (82x) 1.10.0+cpu 3.9.7
Google Colab INTEL Xeon ??? NVIDIA Tesla K80 6.400 с 70.714 мс (163x) 1.10.0+cu113 3.7.12
Kaggle INTEL Xeon ??? NVIDIA Tesla P100 6.626 с 33.878 мс (340x) 1.10.0+cu113 3.7.10

Примечание:

  1. Этот тест был проведён с использованием одной и той же модели Cat or Dog для прогнозирования 10 случайных изображений животных (одинаковых для каждой группы).
  2. Последняя версия PyTorch была вручную установлена на всех платформах, но драйвер и Python остались стандартными. [1]: http://openlabtesting.org:15000/badge?project=pytorch%2Fpytorch&job_name=pytorch-arm64-build-daily-v1.4.0 [2]: https://img.shields.io/badge/build-passing-brightgreen [3]: http://openlabtesting.org:15000/badge?project=pytorch%2Fpytorch&job_name=pytorch-arm64-build-daily-v1.5.0 [4]: https://img.shields.io/badge/build-pending-yellow [5]: http://openlabtesting.org:15000/badge?project=pytorch%2Fpytorch&job_name=pytorch-arm64-build-daily-v1.6.0 [6]: https://img.shields.io/badge/build-running-blue [7]: http://openlabtesting.org:15000/badge?project=pytorch%2Fpytorch&job_name=pytorch-arm64-build-daily-master-py36 [8]: https://img.shields.io/badge/build-failed-red [9]: http://openlabtesting.org:15000/badge?project=pytorch%2Fpytorch&job_name=pytorch-arm64-build-daily-master-py37 [10]: https://github.com/KumaTea/pytorch-aarch64/releases/tag/v1.6.0 [11]: http://openlabtesting.org:15000/badge?project=pytorch%2Fpytorch&job_name=pytorch-arm64-build-daily-master-py38 [12]: https://github.com/KumaTea/pytorch-aarch64/releases/tag/v1.7.0 [13]: https://github.com/pytorch/vision#installation [14]: https://github.com/pytorch/audio#dependencies [15]: https://img.shields.io/badge/build-canceled-lightgrey [16]: https://github.com/KumaTea/pytorch-aarch64/releases [17]: https://pypi.org/project/torchtext/0.6.0/#files [18]: https://github.com/KumaTea/pytorch-aarch64/releases/tag/v1.7.1 [19]: https://status.openlabtesting.org/builds/builds?project=pytorch%2Fpytorch [20]: https://github.com/pytorch/pytorch [21]: https://github.com/pytorch/vision [22]: https://github.com/pytorch/audio [23]: https://linux-sunxi.org/H6 [24]: https://github.com/pytorch/pytorch/issues/33124#issuecomment-602048845 [25]: https://github.com/fastai/fastai [26]: https://www.anaconda.com/products/individual [27]: https://www.fast.ai [28]: https://github.com/pytorch/text [29]: https://github.com/pytorch/text#installation [30]: https://developer.nvidia.com/embedded/jetson-developer-kits [31]: https://gitee.com/kumatea/pytorch-aarch64 [32]: https://github.com/KumaTea/pytorch-aarch64 [33]: https://torch.kmtea.eu [34]: https://hub.docker.com/r/kumatea/pytorch [35]: https://github.com/KumaTea/pytorch-aarch64/releases/tag/v1.5.1 [36]: https://github.com/KumaTea/pytorch-aarch64/releases/tag/v1.5.0 [37]: https://github.com/KumaTea/pytorch-aarch64/releases/tag/v1.4.1 [38]: https://github.com/KumaTea/pytorch-aarch64/releases/tag/v1.4.0 [39]: https://github.com/KumaTea/pytorch-aarch64#official-pytorch-ci-builds [40]: https://travis-ci.org/KumaTea/pytorch-aarch64.svg?branch=main [41]: https://travis-ci.org/KumaTea/pytorch-aarch64 [42]: https://github.com/pytorch/csprng [43]: https://shields.io/badge/python-3.7%20%7C%203.8%20%7C%203.9%20%7C%203.10-blue [44]: https://shields.io/badge/ARM-v7%20%7C%20v8-blue [45]: https://github.com/KumaTea/pytorch-arm [46]: https://github.com/KumaTea/pytorch-aarch64/releases/tag/v1.8.0 [47]: https://ffmpeg.org [48]: https://github.com/KumaTea/pytorch-aarch64/releases/tag/v1.8.1 [49]: https://download.pytorch.org/whl/torch_stable.html [50]: https://circleci.com/gh/KumaTea/pytorch-aarch64.svg?style=svg [51]: https://circleci.com/gh/KumaTea/pytorch-aarch64 [52]: https://github.com/KumaTea/pytorch-aarch64/releases/tag/v1.9.0 [53]: https://github.com/KumaTea/pytorch-aarch64/issues/8 [54]: https://github.com/conda-forge/miniforge [55]: https://github.com/KumaTea/tensorflow-aarch64 [56]: https://github.com/KumaTea/pytorch-aarch64/releases/tag/v1.10.0 [57]: https://github.com/KumaTea/ext-whl [58]: https://anaconda.org/kumatea/pytorch [59]: https://shields.io/badge/ver-1.11.0-brightgreen [60]: https://pypi.org/project/torch/1.8.0/#files

Комментарии ( 0 )

Вы можете оставить комментарий после Вход в систему

Введение

Колёса PyTorch для aarch64 / ARM64, с поддержкой зрения, звука и текста. Развернуть Свернуть
HTML и 3 других языков
MIT
Отмена

Обновления

Пока нет обновлений

Участники

все

Недавние действия

Загрузить больше
Больше нет результатов для загрузки
1
https://api.gitlife.ru/oschina-mirror/kumatea-pytorch-aarch64.git
git@api.gitlife.ru:oschina-mirror/kumatea-pytorch-aarch64.git
oschina-mirror
kumatea-pytorch-aarch64
kumatea-pytorch-aarch64
main