ml-agents环境配置

Unity Machine Learning Agents (ML-Agents) 是一款开源的 Unity 插件, 使得我们得以在游戏环境和模拟环境中训练智能 agent。您可以使用 reinforcement learning(强化学习)、imitation learning(模仿学习)、neuroevolution(神经进化)或其他机器学习方法, 通过简单易用的 Python API进行控制,对 Agent 进行训练。
配置ml-agents环境时,最重要的是要注意cuda,torch,python,mlagents等环境的版本对应,所有对应的版本都可以在GitHub中的文档中查询。

  1. 安装anaconda
  2. 安装cuda和cudnn
  3. 安装unity
  4. 创建项目环境
  5. 安装torch
  6. 安装mlagents的unity部分
  7. 安装mlagents的python部分

1. 安装anaconda

anaconda官网

官网链接:https://anaconda.com/download
官方安装,略。

2. 安装cuda和cudnn

打开cmd,输入nvidia-smi,查看右上角显示的当前驱动支持的cuda最高版本,在nvidia官网下载相应版本的cuda。
在这里插入图片描述

官网链接:https://developer.nvidia.com/cuda-downloads
在这里插入图片描述

下载自己所需要版本的cuda,本文中使用的是cuda12.1版本,下载完成后打开cmd输入nvcc-V,查看是否下载完成。注意下载过程中记住CUDA的安装路径,以便后续查看,默认安装在C盘即可。
在这里插入图片描述

在nvidia官网下载与cuda版本对应的cudnn。
官网链接:https://developer.nvidia.com/rdp/cudnn-download

下载的cudnn是一个压缩包,解压后如图所示,将三个文件夹复制到CUDA目录下,即可完成cudnn的安装。
在这里插入图片描述

3. 安装unity

安装unity部分略过。
注意在安装unity之前,先查看需要安装的mlagents版本对应的unity版本号,一方版本号不对,无法完成后续安装。在github中mlagents官方文档中可以查看当前mlagents版本对应的unity版本号。
在这里插入图片描述

可以看到,在最新版本的mlagents中,需要对应的unity版本是2021.3以及更高版本

4. 创建项目环境

为了让各个项目之间相互不产生干扰,为每个项目创建其单独的环境,若没有创建新的环境,可以在cmd中输入

Conda create -n [你的环境名称] python=X.X

本问中使用的环境名称为ML-agentsDemo,python版本为3.8,注意此处一定不要随意使用最新版本的python,要查看GitHub官方文档当前mlagents版本支持的python版本,本文项目中的最新版本的mlagents支持python3.8,若此处安装更高版本的python,后续过程中会产生很多问题。
创建完新项目的新环境,输入

conda env list

查看环境是否创建成功
在这里插入图片描述

环境创建成功后,输入

Activate [你的环境名称]

进入该环境,在该环境下进行后续的安装操作。
在这里插入图片描述

5. 安装torch

登录pytorch官网,根据自己对应的cuda版本,使用conda下载对应版本的torch,官网会有下载语句,按照官网下载即可。
官网链接:https://pytorch.org
在这里插入图片描述

本文中的环境是cuda12.1,所以选择相应的语句,将该语句在项目环境下输入,即可安装对应版本的pytorch。此过程较为漫长。安装完成后如下图所示。
在这里插入图片描述
安装完成后cmd输入

pyhon
import torch
torch.cuda.is_available()

检查是否安装成功

6. 安装mlagents的unity部分

进入github中mlagents项目首页,找到右侧的最新版本,本文中最新版本为release-20
在这里插入图片描述
官方链接:*https://github.com/Unity-Technologies/ml-agents*
进入对应版本之后可以看到有以下几个包,将整个zip文件下载
在这里插入图片描述下载完成之后得到以下几个包,将其中的com.unity的两个包导入到unity中在这里插入图片描述
打开unity,新建一个项目,在package manager中点击左上角加号,将两个包导入进去
在这里插入图片描述

7. 安装mlagents的python部分

打开cmd,cd进入项目中ml-agents的文件夹
在这里插入图片描述

执行命令

Pip install -e ./

完成安装

  • 3
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 10
    评论
1. 安装Anaconda环境 Anaconda是一个开源的Python发行版本,内置了许多常用的Python包和工具,如numpy、matplotlib等。我们可以通过Anaconda来管理我们的Python环境,方便我们对不同的项目使用不同的Python环境。 下载Anaconda:https://www.anaconda.com/products/distribution 安装Anaconda:下载安装包后,直接运行安装程序,按照提示完成安装即可。 2. 创建虚拟环境 在Anaconda中,我们可以通过创建虚拟环境来管理我们的Python环境。因为ml-agents需要使用UnityPython环境,我们需要创建一个与Unity相同的Python环境。 打开Anaconda Prompt终端,输入以下命令: ``` conda create -n unity python=3.6 ``` 这条命令会创建一个名为"unity"的虚拟环境,并安装Python 3.6版本。我们可以根据需要选择其他Python版本。 3. 激活虚拟环境 在创建完虚拟环境后,我们需要激活它才能使用。在Anaconda Prompt终端中,输入以下命令: ``` conda activate unity ``` 这条命令会激活名为"unity"的虚拟环境。 4. 安装ml-agents 在激活虚拟环境后,我们可以通过以下命令安装ml-agents: ``` pip install mlagents ``` 这条命令会安装ml-agents及其依赖的包。 5. 安装Unity ml-agents需要与Unity一起使用,因此我们需要先安装Unity。 下载Unity:https://store.unity.com/download-nuo 安装Unity:下载安装包后,直接运行安装程序,按照提示完成安装即可。 6. 配置Unity 安装完Unity后,我们需要在Unity中导入ml-agents的插件。具体操作如下: 在Unity中创建一个新项目,选择2D或3D模板即可。 打开菜单栏"Window"->"Package Manager",在搜索框中输入"ml-agents",然后点击"Install"按钮进行安装。 安装完成后,打开菜单栏"Window"->"AI"->"ML-Agents"->"Agent Settings",在"Python"中选择我们刚才创建的虚拟环境路径,例如: ``` C:\Users\your_username\Anaconda3\envs\unity\python.exe ``` 7. 测试ml-agents环境 在安装和配置完毕后,我们可以进行简单的测试,验证ml-agents是否安装成功。 在Unity中,打开菜单栏"Window"->"AI"->"ML-Agents"->"Create New Agent",创建一个新的Agent。 添加一个"Behavior Name"为"test"的Behavior。 在"Behavior Parameters"中,将"Behavior Type"设置为"Default". 在"Behavior Parameters"中,将"Observation Shapes"设置为"Vector Observation",并将"Vector Observation Size"设置为1。 在"Behavior Parameters"中,将"Action Space Type"设置为"Discrete",并将"Discrete Branches"设置为2。 在"Behavior Parameters"中,将"Output"设置为"Action". 点击"Save"按钮保存。 在菜单栏"File"->"Save Project",保存Unity项目。 在Anaconda Prompt终端中,进入Unity项目的根目录,例如: ``` cd C:\Users\your_username\Documents\Unity Projects\test ``` 在终端中,输入以下命令: ``` mlagents-learn config\test.yaml --run-id=test --train ``` 这条命令会启动ml-agents的训练过程。 在训练过程中,我们可以在Unity中查看Agent的行为。在菜单栏"Window"->"AI"->"ML-Agents"->"Agent Monitor"中,选择刚才创建的Agent和Behavior,然后点击"Open"按钮打开Agent Monitor。 在Agent Monitor中,我们可以观察Agent的行为,以及训练过程中的reward等信息。 如果能够成功观察到Agent的行为,并且训练过程中reward等信息正常,说明ml-agents环境安装成功。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 10
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值