跳转到主要内容

快速轻松地运行《星际争霸:母巢之战》作为Gym环境的便捷方式。

项目描述

TorchCraft 2 CircleCI

(又称StarCraft Gym)

使用星际争霸作为强化学习环境的最快、最简单的方式。

要开始使用,您只需pip install即可。

动机

研究人员迫切希望使用基于星际争霸的环境进行强化学习实验。但是,将星际争霸设置并运行起来可能非常耗时,并且大多数环境都不易直接使用。

TorchCraft 2提供简单(一键!)安装、易于使用的Gym接口和开箱即用的多个挑战/基线。

此外,我们与Blizzard合作,允许TorchCraft 2包含首次合法分发《星际争霸:母巢之战》的二进制文件。这将使公众能够使用TorchCraft 2,而无需获取自己的《星际争霸》副本或安装它。

有关详细信息,请参阅项目提案

路线图

有了这些准备,我们旨在内部启动TorchCraft 2进行实验和beta测试。从那里开始,计划是

  • 与TorchCraft集成(弃用原始Lua/Python API)
  • 在PyPy上提供预编译的二进制文件分发
  • 公开发布、PR和教程内容

完整的路线图位于GitHub

使用方法

为开发构建TorchCraft 2

这是我现在推荐使用TorchCraft 2的方法

git clone --recursive https://github.com/fairinternal/TorchCraft2/ tc2
cd tc2

conda create --name tc2 python=3 pip
source activate tc2
conda install pip cmake pybind11 numpy
conda install -y -c conda-forge sdl2 zstd
conda install -y -c anaconda zeromq
pip install -e .

下载所需的星际争霸文件

一旦您安装了tc2,您就必须下载星际争霸数据文件(MPQ文件)。我们为您提供了一个专门为您使用的工具 - 我们无法立即提供它们,因为我们需要您先阅读并接受Blizzard的EULA。

tc2-setup

演示

本演示(run_demo.py)创建了一个星际争霸健身房环境,并使用简单的“攻击中部”策略控制智能体。

import gym
import tc2
from tc2.agents import attack_middle

env = gym.make('tc2-demo-v0')
while True:
  env.reset()
  actions = []
  done = False
  while not done:
    observation, reward, done, info = env.step(actions)
    actions = attack_middle(observation)

运行 TorchCraft 2 演示

  • python3 run_demo.py

项目详情


下载文件

下载适用于您平台的应用程序。如果您不确定选择哪个,请了解有关安装包的更多信息。

源分发

tc2-0.0.1594993188.tar.gz (8.4 MB 查看散列

上传时间

由以下组织支持

AWS AWS 云计算和安全赞助商 Datadog Datadog 监控 Fastly Fastly CDN Google Google 下载分析 Microsoft Microsoft PSF赞助商 Pingdom Pingdom 监控 Sentry Sentry 错误日志 StatusPage StatusPage 状态页面