Unity ML-agents怎么下载安装

ML-agents

1.创建venv

 

1).在ML-agents II文件夹下面输入cmd打开在该路径下的cmd，然后输入python -m venv venv

2).然后我们就可以在文件列表看到venv文件

 

2.激活环境

1).在cmd中输入venv\Scripts\activate激活虚拟环境

2).更新确保pip处于最新版本

3.安装torch

如果直接输入pip install torch会自动下载一个torch，但是这个torch是cpu版本的，我们需要gpu版的

 

(这里cpu版的只有172MB，而gpu版是几个GB的)

 

从官网上找到的是这个，我们使用清华源下载：

pip3 install torch torchvision torchaudio -i https://pypi.tuna.tsinghua.edu.cn/simple --index-url https://download.pytorch.org/whl/cu117

(如果你有全局的梯子那么直接按照官网的下载即可)

因为我使用的是python3.10，所以我下载CUDA 11.7就足够了

 

4.安装mlagents

 

输入pip install mlagents以下载

安装完成后输入mlagents.exe -learn报错，猜测是python版本太高，于是使用python3.9重新部署。

但是同样遇到了刚才那样的报错，于是询问chatgpt得知是：

根据错误信息，问题出现在protobuf库中的message.Message.CheckCalledFromGeneratedFile()函数。这个错误通常是由于protobuf版本与其他依赖库的版本不兼容导致的。

要解决这个问题，建议你尝试降低protobuf库的版本到3.20.x或更低，因为在3.20.x版本之后，protobuf库对于Python生成的protobuf代码进行了一些更改，可能导致与其他库不兼容。

你可以使用以下命令来降低protobuf库的版本：pip install protobuf==3.20.0

接下来成功解决问题。

(接下来为尝试python3.10可以完成不?) (yes!)

5.指定使用GPU

由于我们使用的是cuda版的torch，因此在help的末尾可以看到

 

这要求我们指定是使用cpu还是gpu，是哪个gpu？

当我们输入mlagents-learn.exe --torch-device cuda如果显示如下，那么我们就是成功了。

 

下篇文章将从unity中介绍如何使用ml-agents