快捷方式

TorchRL

_images/logo.png

TorchRL 是一个用于 PyTorch 的开源强化学习 (RL) 库。

您可以直接从 PyPI 安装 TorchRL(有关安装说明的更多信息,请参阅下面的专用部分)

$ pip install torchrl

TorchRL 提供以 pytorch 和 python 为优先的低级和高级 RL 抽象,旨在实现高效、模块化、文档完善和经过充分测试。其代码旨在支持 RL 研究。大部分代码以高度模块化的方式用 python 编写,以便研究人员可以轻松地交换组件、对其进行转换或轻松编写新组件。

此仓库试图与现有 pytorch 生态系统库保持一致,因为它具有“数据集支柱”(环境)转换模型、数据工具(例如收集器和容器)等。TorchRL 旨在尽可能减少依赖项(python 标准库、numpy 和 pytorch)。常见的环境库(例如 OpenAI gym)仅为可选。

在低级方面,torchrl 附带了一组高度可重用的函数,用于成本函数回报和数据处理。

TorchRL 旨在实现高模块化和良好的运行时性能。

要了解有关 TorchRL 理念和 API 参考之外的功能的更多信息,请查阅TorchRL 论文

安装

TorchRL 的发布与 PyTorch 同步,因此请确保您始终使用最新版本的 PyTorch 来享受该库的最新功能(尽管核心功能保证向后兼容 pytorch>=2.0)。每晚发布版本可以通过以下方式安装

$ pip install tensordict-nightly
$ pip install torchrl-nightly

或通过 git clone,如果您愿意为该库做出贡献

$ cd path/to/root
$ git clone https://github.com/pytorch/tensordict
$ git clone https://github.com/pytorch/rl
$ cd tensordict
$ python setup.py develop
$ cd ../rl
$ python setup.py develop

入门

一系列快速教程,帮助您掌握该库的基本功能。如果您时间有限,可以从该系列的最后一项开始,并在需要了解更多信息时导航到前面的教程!

教程

基础知识

中级

高级

参考资料

知识库

索引和表格

文档

访问全面的 PyTorch 开发者文档

查看文档

教程

为初学者和高级开发者提供深入的教程

查看教程

资源

查找开发资源并让您的问题得到解答

查看资源