tianshou

thu-ml 在 GitHub 开源的算法库，主要语言 Python。 An elegant PyTorch deep reinforcement learning library.

SDKthu-mltianshou

开发工具档案怎么用、适配哪些机型、哪些环境验证过

thu-ml 在 GitHub 开源的算法库，主要语言 Python。 An elegant PyTorch deep reinforcement learning library.

算法库python更新于 6 个月前

概述

这个开发工具的核心职责、所属技术栈与硬件生态定位。

tianshou是清华大学THU-ML团队开源的深度强化学习算法库，属于Python生态下的算法库类工具，基于PyTorch框架开发，核心功能覆盖深度强化学习全流程的算法实现、训练调度、结果验证等模块，具备简洁高效的API设计与良好的可扩展性，可支撑强化学习相关的算法研发与落地验证需求。

该库适配机器人技术栈的高层决策、策略学习层级，可对接运动控制、VLA策略训练、机器人仿真环境交互等上层开发场景，开发者可通过Python包管理工具直接安装导入，依托其预置的各类强化学习算法接口快速构建训练 pipeline，适配各类机器人相关的强化学习训练需求。

工具分类、开发语言、GitHub 社区活跃度与环境要求。

源码克隆、构建方式、已验证系统环境与版本更新记录。

git clone https://github.com/thu-ml/tianshou.git
cd tianshou
# 安装 Python 依赖与开发包
pip install -e .

版本 / 分支	验证环境	接入说明
main / 最新稳定分支（适配PyTorch 2.x系列版本）	Ubuntu 22.04 LTS / Python 3.8~3.11 / PyTorch 2.0+ / CUDA 11.7及以上	可通过pip install tianshou直接安装，核心入口为tianshou.trainer、各算法实现模块，支持对接Gym等仿真环境开发

支持接入该 SDK、ROS 驱动或仿真资产的机器人本体。

尚未收录适配机型。

同一厂商开源的配套仓库或同技术栈开发工具。

本页事实对应的公开页面，以原文为准。