Установка PuzzleLib на Windows

Общие требования

Перед тем, как установить сам PuzzleLib, требуется установить следующие системные зависимости:

  • С++ компилятор (cl из Visual Sudio >= 2015, gcc, clang); путь к папке с компилятором нужно прописать в PATH (путь может быть, например, C:\Program Files (x86)\Microsoft Visual Studio ...\VC\bin для cl.exe);
  • python >= 3.5 (64 битный).

Дополнительные требования

CUDA (NVIDIA GPU)

Для работы с видеокартами NVIDIA требуются:

DNNL (Intel|AMD CPU)

  • DNNL одной из версий [1.1, 1.2, 1.91]

Для библиотеки dnnl.dll из архива необходимо указать путь к файлу в системной переменной DNNL_PATH.

Некоторые функции библиотеки, которые не поддерживаются DNNL, будут выполнятся на numpy-бэкенде.

TensorRT (только инференс для NVIDIA GPU)

Путь к файлам TensorRT необходимо прописать в системной переменной TRT_PATH, иначе TensorRT будет предполагаться установленным по пути системной переменной CUDA_PATH.

OpenVINO (только инференс для Intel CPU)

Путь к файлам OpenVINO необходимо прописать в системной переменной OPENVINO_PATH, иначе ошибка сборки.

Установка из PyPI

Для установки версии, работающей на чистом numpy бэкенде и имеющей урезанный функционал:

$ pip install PuzzleLib

Также на выбор есть две опции:

  • backend - какой дополнительно бэкенд поставить; возможные варианты - cuda, intel;
  • converter - какой конвертер подготовить для ускоренного инференса; возможные варианты - tensorrt, openvino.

Например:

установка cuda и intel бэкендов:

$ pip install PuzzleLib --install-option="--backend=cuda,intel"

установка cuda бэкенда, а также tensorrt конвертера:

$ pip install PuzzleLib --install-option="--backend=cuda" --install-option="--converter=tensorrt"

Установка из исходников

Выберите и скачайте нужный релиз на странице релизов.

Распакуйте архив по выбранному вами пути (path/to/PuzzleLib).

Установить библиотеку можно прямым вызовом скрипта setup.py:

$ python /path/to/PuzzleLib/setup.py install --backend=cuda

Либос помощью pip:

$ pip install /path/to/PuzzleLib --install-option="--backend=cuda"

См. Установка из PyPI для ознакомления с возможными опциями установки.

Tip

Изменения системных переменных консоли становятся видны только после перезапуска консоли, поэтому, если у вас что-то не работает - попробуйте закрыть-открыть консоль.

Выбор бэкенда

>>> from PuzzleLib import Config
>>> Config.backend = Config.Backend.cuda # для использования NVIDIA CUDA бэкенда
>>> Config.backend = Config.Backend.hip # для использования AMD ROCm бэкенда
>>> Config.backend = Config.Backend.cpu # для использования CPU numpy бэкенда
>>> Config.backend = Config.Backend.intel # для использования Intel/AMD CPU DNNL бэкенда