Unity ML-Agents の Python環境を一括でセットアップする

概要

UnityでML-Agentsを利用する際に毎回行うPython環境の構築が面倒だったので
ダウンロードから仮想環境構築まで一括で行うシェルスクリプトを作りました。
仮想環境を作りたいフォルダ内でsource setup.shのように実行してください。
※現在のML-Agentsの最新リリースではPython3.10.x以上を使う必要があります。
ML-Agents公式GitHub

シェルスクリプトの内容

#!bin/zsh

python --version
echo "Start mlenv setup"
python -m venv env
echo "venv is created"

TAG=$(curl -s https://api.github.com/repos/Unity-Technologies/ml-agents/releases/latest \
| grep "tag_name" \
| cut -d : -f 2,3 \
| tr -d ' ' \
| tr -d \" \
| tr -d ,)

echo "Download ${TAG}"
curl -LO https://github.com/Unity-Technologies/ml-agents/archive/refs/tags/${TAG}.zip
echo "Download is done"

unzip ${TAG}.zip
echo "unzip is done"
rm ${TAG}.zip
echo "zip file is removed"

source env/bin/activate
echo "venv is activated"

pushd ml-agents-${TAG}
pip install -e ./ml-agents-envs
pip install -e ./ml-agents
popd

deactivate

echo "Setup is done"