一、前期准备

1. 通过Unity Hub安装Unity 2023.2或更高版本（需注册Unity账号），本文安装了 Unity 2023.2.13f1 。之前装了 Unity 6 和 2022 的，已踩坑，显示版本不匹配，无法打开 ML-agents（release 22）的官方demo文件。
   

2. 已安装NVIDIA驱动程序，在终端中通过使用 nvidia-smi 确认驱动版本安装及 CUDA 版本。

3. 推荐使用Anaconda进行虚拟环境管理，并修改好环境变量，确保conda list 可以正常输出，正常创建、激活和删除虚拟环境。后续在安装 ML-agents时踩坑后，直接删除环境重建即可。

前期准备中的内容在CSDN上已有许多教程，在此不再赘述。主要针对在Ubuntu22.04 安装 Unity 的 ML-agents 工具包（release 22 版本）。

二、官方教程链接

Github： https://github.com/Unity-Technologies/ml-agents
Getting Started Guide：https://github.com/Unity-Technologies/ml-agents/blob/develop/docs/Getting-Started.md
安装指引：https://github.com/Unity-Technologies/ml-agents/blob/develop/docs/Installation.md

官方安装教程感觉写得有点冗长，而且也踩了一些坑，特此整理。

三、梳理安装过程

1. 使用Anaconda创建独立环境，并激活环境，后续操作均在该虚拟环境中：

在终端中创建环境，并激活。
※ 特别注意，ML-agents（release 22 版本）的python版本官方推荐为3.10.10~3.10.12。
本文采用3.10.12成功安装。若只指定 python=3.10，可能会我安装其他小版本，只能删除环境重来，我踩过雷。

conda create -n mlagents python=3.10.12
conda activate mlagents

2. 克隆ML-Agents仓库，并打开克隆的 ml-agents 文件夹：

如后续版本更新，可前往Github仓库修改 git clone 链接
官方Github仓库： https://github.com/Unity-Technologies/ml-agents

git clone --branch release_22 https://github.com/Unity-Technologies/ml-agents.git
cd ml-agents

3. 安装系统级编译依赖：

确保Python头文件及构建工具就绪：

sudo apt-get install python3-dev build-essential

4. PyTorch的安装与验证：

4. 1 安装PyTorch：

已下代码针对本机的CUDA 12.1环境（需提前安装NVIDIA驱动）：

pip install torch==2.2.1 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

如GPU、NVIDIA驱动、CUDA版本与本文不一致，请自行查询匹配关系
或将你的硬件信息和驱动信息输入询问Deepseek（亲测好用）。
了解对应关系后，前往Pytorch官网进行安装。
PytTorch官网： https://pytorch.org/get-started/locally/

4.2 进行PyTorch验证：

在 mlagents 虚拟环境中，输入：

pip list | grep -E "torch|vision|audio"

我的输出，类似输出即完成PyTorch安装：

5. 安装ML-agents相关的Python包：

pip install -e ./ml-agents-envs
pip install -e ./ml-agents

6. 在Unity软件中（需要Unity 2023.2或更高版本。本文为 Unity 2023.2.13f1）集成 ML Agents 和 ML Agents Extension 包：

① 在Unity Editor中：
② 打开顶部菜单项 Window > Package Manager
③ 点击 + 选择 Add package from disk
④ 导航至克隆仓库（即步骤2中的ml-agents文件夹）的 com.unity.ml-agents/package.json 文件
⑤ 以相同方法，添加安装克隆仓库（即步骤2中的ml-agents文件夹）的 com.unity.ml-agents.extensions/package.json 文件
⑥ 完成安装读条后，ML Agents 与 ML Agents Extension 均显示Local，如图所示

7. 在终端对应虚拟环境中，验证安装：

终端执行以下命令确认 mlagents 的 Python包安装成功：

mlagents-learn --help

输出如下即安装成功：