Установка bosk

Пожалуйста, внимательно прочитайте эту инструкцию, так как пакет bosk зависит от библиотеки JAX, которая не может быть установлена в автоматическом режиме.

Перед установкой

Окружение

Убедитесь, что Вы создали изолированное окружение для python версии не ниже 3.9.

Например, с помощью Anaconda Вы можете создать и активировать окружение, выполнив:

conda create -n bosk_env python=3.10
conda activate bosk_env

Установка Graphviz

Bosk использует библиотеку Graphviz для визуализации вычислительных графов. Для корректной работы данного инструмента требуется установка бинарной зависимости graphviz.

В Linux graphviz можно установить с помощью пакетного менеджера. Например, в Ubuntu может быть использован следующий код:

sudo apt install graphviz

В случае использования Anaconda, graphviz может быть установлен с помощью команды:

conda install -c conda-forge graphviz

Для прочих платформ инструкции по установке можно найти по ссылке.

Установка JAX

Если необходимо установить bosk без поддержки JAX, перейдите к установке пакета.

Bosk использует JAX для вычислений на графическом процессоре, но установка JAX не тривиальна. Официально JAX распространяется только для Linux и Mac OS, поэтому, к сожалению, пользователям Windows следует использовать WSL для установки JAX и использования bosk с поддержкой вычислений на ГПУ.

Установка только для ЦПУ

Если на Вашей системе нет графического процессора, Вы можете установить версию JAX для ЦПУ:

pip install --upgrade "jax[cpu]==0.4.10"

Установка с поддержкой вычислений на ГПУ

Пожалуйста, обратитесь к официальной инструкции по установке JAX или выполните следующую команду для установки версии с поддержкой CUDA12:

pip install --upgrade "jax[cuda12_pip]==0.4.10" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html

Note

Обратите внимание, что, согласно руководству NVIDIA, пользователи Windows не должны устанавливать какой-либо драйвер NVIDIA GPU Linux внутри WSL 2. Участники NTAILab, использующие Windows, установливали CUDA Toolkit и cuDNN вручную, поэтому мы можем рекомендовать только часть руководства, где предустановленная копия CUDA используется для установки JAX поверх нее.

Установка пакета

Для установки пакета bosk напрямую из GitHub выполните:

pip install git+https://github.com/NTAILab/bosk.git

Также Вы можете вручную скопировать репозиторий и установить bosk:

git clone https://github.com/NTAILab/bosk.git
cd bosk
pip install -r requirements.txt
python setup.py install

Установка в режиме разработчика

Если Вы планируете вносить изменения в bosk, будет удобнее установить пакет в режиме разработчика, чтобы Python сам регистрировал вносимые изменения при запуске:

git clone https://github.com/NTAILab/bosk.git
cd bosk
pip install -r requirements.txt
python setup.py develop