在线时间:8:00-16:00
迪恩网络APP
随时随地掌握行业动态
扫描二维码
关注迪恩网络微信公众号
开源软件名称:DI-hpc开源软件地址:https://gitee.com/opendilab/DI-hpc开源软件介绍:DI-HPC: Decision Intelligence - High Performance ComputationDI-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
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 StartInstall from whlThe easiest way to get DI-HPC is to use pip, and you can get
and then call $ pip install <YOUR_WHL> Install from source codeAlternatively you can install latest DI-HPC from git master branch: $ python3 setup.py install Run on LinuxYou will get benchmark result by following commands: $ python3 tests/test_gae.py |
请发表评论