Unity ML-Agents系列教程（一）环境安装

最新推荐文章于 2025-07-09 13:00:51 发布

原创最新推荐文章于 2025-07-09 13:00:51 发布 · 1.4k 阅读

14 ·

CC 4.0 BY-SA版权

文章标签：

#unity #游戏引擎 #人工智能 #机器学习

unity3d 同时被 3 个专栏收录

146 篇文章

订阅专栏

游戏开发踩坑记录

137 篇文章

订阅专栏

ML-Agent系列教程

1 篇文章

订阅专栏

（一）背景

最近人工智能有点火，各种机器人、大模型等，所以也学习下Unity下的人工智能，看看Unity怎么机器学习的。

因为ML Agent这个是多年前Unity弄的，所以现在找到的资料都是几年前的，而现在Python和其他依赖库都已经升级n个版本了，所以按照现在找到的资料大部分都是报错。

后来经过踩坑好久，各种报错解决，终于找到一个靠谱的了，部署成功了。（出现Unity Logo说明成功了）

以上是背景，下面说下详细步骤：

（二）安装Conda

如果你刚接触肯定对各种都很陌生，首先Conda是什么，Conda 是一个开源的软件包管理系统和环境管理系统，通过Conda可以很方便的创建虚拟环境，在各个虚拟环境中切换，不会导致相互环境冲突。下图就是各种虚拟环境。

到Conda官网Conda Documentation — conda 25.3.0 documentation下载对应的安装包即可，我目前是在windows里的，所以下载windows的conda安装包。

下载完在这里能看到Anaconda Prompt，这个软件后面我们用来输入命令行。

Conda 安装好之后已经包含 Python了，就不需要单独再下载Python了。

（三）准备Unity ML-Agent官方示例包

我们先把官方示例跑通，官方给了很多示例来熟悉这个框架。

进入Unity的官方示例GitHub，将工程下载下来或者clone下来。GitHub - Unity-Technologies/ml-agents: The Unity Machine Learning Agents Toolkit (ML-Agents) is an open-source project that enables games and simulations to serve as environments for training intelligent agents using deep reinforcement learning and imitation learning.

其中，Project就是Unity的工程，我们用Unity打开即可，官方提供了很多示例，我们直接运行也可以，但是直接运行是用训练好的模型跑的，我们后面再来说下怎么自己训练模型。

（四）安装Python环境

通过下面的命令分别在Conda里敲，如果链接太慢，可以把--index-url改成阿里云的镜像：https://mirrors.aliyun.com/pypi/simple/

也就是pip3 install torch -i https://mirrors.aliyun.com/pypi/simple/

备注：后面-i就是--index-url的缩写，-i指定了从哪里安装库。

# 1、使用Conda创建一个虚拟环境，-n后面的mlagents就是虚拟环境名称，指定Python版本
conda create -n mlagents python=3.10.12

# 2、激活虚拟环境
conda activate mlagents

# 3、使用下面命令安装依赖库
conda install numpy=1.23.5
pip3 install torch~=2.2.1 --index-url https://download.pytorch.org/whl/cu121

# 4、测试下安装成功没
python
import torch
import numpy
print(torch.__version__)
print(numpy.__version__)
exit()

# 5、清空命令行内容（可选）
clear

# 6、CD到clone下来的文件夹（这个你们下载在哪里就改成哪里，根目录）
cd /d E:\360Downloads\ml-agents-latest_release\ml-agents-latest_release

# 7、从clone下来的工程里安装依赖项
python -m pip install ./ml-agents-envs
python -m pip install ./ml-agents

# 8、检查训练库安装成功没
mlagents-learn --help

# 9、开始启动一个训练项，供Unity连接，当前为Basic示例
mlagents-learn config/ppo/Basic.yaml --run-id=run1

# 强制覆盖或恢复上一次运行（可选）
mlagents-learn config/ppo/Basic.yaml --run-id=run1 --force
mlagents-learn config/ppo/Basic.yaml --run-id=run1 --resume

mlagents-learn --help会出险unity的logo，说明安装成功了。