环境配置主要参考:
python侧环境搭建
安装python(版本要求高于3.6,详见文档)
在unity项目下新建python虚拟环境,window下命令行如下
cd unityprojectpath
python -m venv venv
venv\Scripts\activate
- 更新pip,安装ml模块
python -m pip install --upgrade pip
pip install torch
pip install mlagents
在此过程中如果遇到依赖模块的版本与mlagents所需版本不一致,则指定resolver来解决依赖问题
pip install mlagents --use-feature=2020-resovler
- 安装cuda
unity侧环境搭建
unity侧环境只需要package manager内开启priview package
即可