ML-Agents 在Mac M1上的安装


前言

提示:这里可以添加本文要记录的大概内容:

最近为了开发需要,想着把老苹果笔记本换掉买一台新的苹果电脑,因为老电脑开发Unity实在是有点太卡了,入手前就知道新的苹果笔记本M1芯片可能存在很多坑,但多年的苹果系统使用已经让我再也提不起兴趣使用windows了,因为在使用windows虚拟机(工作需要)的时候往往安装个软件会有很多广告啥的。因此,我果断割肾买了台苹果笔记本,没想到这次入的坑有点深。。。Unity在苹果电脑上也跑出了一堆问题,好多插件不支持,什么Obi,AVProMovieTexture啥的。。。

扯的有点远了,本次主要是解决在苹果电脑上强化学训练问题,安装这个也是搞了半天。。。不过最终还是被我搞定了。参考了博主CottePutman的介绍,但是很可惜,我按照步骤一步一步的实现,最终还是失败了。不过最后还是让我找到了办法,因此记录一下,希望能够帮助到要在Mac M1上安装ML-Agents的的小伙伴们。


一、ML-Agents是什么?

在这里我就不解释了,想安装的人应该都知道。

二、我的Mac系统配置

设备版本
MacBook Pro M1 Max 202164G内存
MacOS MontereyVersion 12.3

三、安装过程

1. MiniConda 安装

参考烟缈睿晨的文章安装 Miniconda。成功后重启终端,你应该能看到前面有(base)字样,表示你的 Miniconda 已经可以正常使用了。

2. ML-Agents 安装:

简易安装方法:

提示:我发现Ml-Agents以及相关东西更新速度挺快的,本安装方法在发文的时候刚刚测试通过,但是后期是否依旧可以就不确定了哈!(^^)
经过了大量的尝试,最简单的安装方式如下:

conda create -n ml-agents python=3.9.7 -y
conda activate ml-agents
conda install pytorch=1.8 -y
conda install grpcio h5py -y
pip install mlagents==0.28.0 -q

复杂安装方法:

问题很多的小明就要说啦,我非要从源码来安装可以吗?答案是肯定的,但过程肯定是复杂的。先按照博主CottePutman的方法,创建一个python3.8的环境,并且进入环境。

conda create -n test python=3.8
conda activate test

然后还是回到 Pytorch 下载网址,下载 1.9.0 版本的 Pytorch,文件名为:

cpu/torch-1.9.0-cp38-none-macosx_11_0_arm64.whl

修改Ml-agents 对PyTorch的支持,打开ml-agents-main/ml-agents/setup.py文件,我们能看到 ML-Agents 对 Pytorch 版本的要求如下:

"torch>=1.8.0,<1.9.0;(platform_system!='Windows' and python_version>='3.9')",
"torch>=1.6.0,<1.9.0;(platform_system!='Windows' and python_version<'3.9')",

修改为:

"torch>=1.8.0,<1.9.0;(platform_system!='Windows' and python_version>='3.9')",
"torch>=1.6.0,<=1.9.0;(platform_system!='Windows' and python_version<'3.9')",

然后安装刚下载的pytorch

pip install  ~/Downloads/torch-1.9.0-cp38-none-macosx_11_0_arm64.whl

关键来了,这边和博主不一样,不是用pip 来安装H5PY,而是按照上面的方式来安装,如下:

conda install grpcio h5py -y

上面步骤做完以后,就可以安心安装ml- agents了,找到下载ml-agents的路径,然后执行如下即可:

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

至此,ml-agents的两种安装方式就这样啦!


总结

我在之前的苹果电脑上做过训练,感觉速度很慢,但是这次使用M1训练明显感觉速度提升了不少,希望能对想在Mac M1上训练的小伙伴能有写帮助吧。

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值