单卡RTX 3080 Ti训练AI世界模型通关《宝可梦 红》
开发者stmonty用单张RTX 3080 Ti训练约1250万参数世界模型,在《宝可梦 红》中成功选出杰尼龟,代码已开源。
一位名为stmonty的开发者仅用一张RTX 3080 Ti游戏显卡,训练了一个小型AI世界模型来游玩《宝可梦 红》。该模型参数量约为1250万,相关技术说明于9月20日发布。
训练数据来自游戏存档中奥克博士实验室场景,共使用42,382帧灰度画面,覆盖1000多次短程运行。模型并不预测完整的下一帧画面,而是采用压缩后的192个数字摘要作为输入。
经过微调后,该模型成功选择了杰尼龟作为初始宝可梦。在从同一存档出发的100多次运行中,有52次计划成功获得初始宝可梦;作为对比,随机按键为0次,未训练模型为1次。
项目代码已在GitHub上以lePokeRed名称开源,推荐使用支持CUDA的GPU运行。据tomshardware.com报道,这一实验展示了在消费级硬件上训练世界模型的可行性。
世界模型研究背景
2026年3月,一篇论文介绍了一个约1500万参数的单一JEPA模型,可在单张GPU上训练。此前,Yann LeCun离开Meta,转而通过其AMI Labs构建世界模型。
快速问答
这个AI模型用了多少参数?
约1250万参数,在单张RTX 3080 Ti上训练。
模型成功选出了哪个初始宝可梦?
经过微调后,模型成功选择了杰尼龟。
代码在哪里可以获取?
项目代码已在GitHub上以lePokeRed名称开源,推荐使用支持CUDA的GPU。