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

OSCHINA-MIRROR/opendilab-DI-hpc

Присоединиться к Gitlife
Откройте для себя и примите участие в публичных проектах с открытым исходным кодом с участием более 10 миллионов разработчиков. Приватные репозитории также полностью бесплатны :)
Присоединиться бесплатно
Клонировать/Скачать
README.md 1.7 КБ
Копировать Редактировать Web IDE Исходные данные Просмотреть построчно История
gitlife-traslator Отправлено 30.11.2024 09:56 d6f2fdb

DI-HPC: Decision Intelligence — High Performance Computation

DI-HPC — это компонент оператора ускорения для общих алгоритмических модулей в алгоритмах обучения с подкреплением, таких как GAE, n-step TD и LSTM и т. д. Операторы поддерживают прямое и обратное распространение и могут использоваться в обучающих модулях, модулях сбора данных и тестирования.

Требования:

  • CUDA 9.2;
  • PyTorch 1.5 (рекомендуется);
  • Python 3.6.9;
  • платформа Linux.

Примечание: рекомендуется использовать одну и ту же среду для DI-HPC и DI-Engine, при этом подойдёт версия PyTorch от 1.3.1 до 1.8.

Быстрый старт

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

Самый простой способ получить DI-HPC — использовать pip, и вы можете получить .whl из:

  • di_hpc_rll-0.0.2-cp36-cp36m-linux_x86_64.whl;
  • di_hpc_rll-0.0.2-cp37-cp37m-linux_x86_64.whl;
  • di_hpc_rll-0.0.2-cp38-cp38-linux_x86_64.whl.

Затем вызовите:

$ pip install <YOUR_WHL>

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

В качестве альтернативы вы можете установить последнюю версию DI-HPC из основной ветки git:

$ python3 setup.py install

Запуск на Linux

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

$ python3 tests/test_gae.py

Опубликовать ( 0 )

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

1
https://api.gitlife.ru/oschina-mirror/opendilab-DI-hpc.git
git@api.gitlife.ru:oschina-mirror/opendilab-DI-hpc.git
oschina-mirror
opendilab-DI-hpc
opendilab-DI-hpc
main