README.md

    DI-HPC: Decision Intelligence - High Performance Computation

    DI-HPC is an acceleration operator component for general algorithm modules in reinforcement learning algorithms, such as GAE, n-step TD and LSTM, etc. The operators support forward and backward propagation, and can be used in training, data collection, and test modules.

    Requirements

    • CUDA 9.2
    • PyTorch 1.5 (recommend)
    • python 3.6.9
    • Linux Platform

    Note: We recommend that DI-HPC and DI-Engine share the same environment, and it should be fine with PyTorch from 1.3.1 to 1.8.

    Quick Start

    Install from whl

    The easiest way to get DI-HPC is to use pip, and you can get .whl from

    and then call

    $ pip install <YOUR_WHL>

    Install from source code

    Alternatively you can install latest DI-HPC from git master branch:

    $ python3 setup.py install

    Run on Linux

    You will get benchmark result by following commands:

    $ python3 tests/test_gae.py

    项目简介

    OpenDILab RL HPC OP Lib

    发行版本

    当前项目没有发行版本

    贡献者 2

    Z zhangming @zhangming
    O opendilab @opendilab

    开发语言

    • Python 56.5 %
    • C++ 23.7 %
    • Cuda 19.8 %