Unity AI項目筆記
一、創(chuàng)建虛擬環(huán)境
????在開始進行Unity AI項目之前,首先需要設置一個適當?shù)奶摂M環(huán)境。以下步驟將會指導你如何創(chuàng)建:
1.下載Python 3.7。
2.下載Anaconda 2020.11??梢栽谇迦A大學開源軟件鏡像站找到鏡像資源:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
3.使用Anaconda創(chuàng)建一個新的環(huán)境,命名為ml-agents。如果你使用的是命令行,可以使用以下命令來創(chuàng)建:
conda create --name ml-agents python=3.8
4.在Anaconda Prompt中輸入activate ml-agents來激活你剛創(chuàng)建的環(huán)境。
5.安裝tensorflow 2.2.0,命令:
python -m pip install tensorflow==2.2.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
6.安裝torch 1.7.1,命令:
pip install torch==1.7.1 -i https://pypi.tuna.tsinghua.edu.cn/simple
7.安裝ml-agents,命令:
pip install mlagents -i https://pypi.tuna.tsinghua.edu.cn/simple
(注意:安裝過程中請確保沒有啟動VPN,否則可能會導致安裝失?。?/p>
8.如果訓練過程中報錯,可能需要降級protobuf版本,命令:
pip install protobuf==3.20.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
二、訓練
????在設置好環(huán)境之后,你就可以開始訓練你的Unity AI項目了:
1.在Anaconda Prompt中輸入activate ml-agents來激活你的環(huán)境。
2.輸入E:切換到E盤。(2、3步需要根據(jù)自己項目來)
3.通過cd E:\UnityProjects\TestAI\Assets\Train進入你項目中的訓練目錄。
4.輸入mlagents-learn config.yaml開始訓練。
如果需要恢復之前的訓練,可以輸入mlagents-learn config.yaml --resume。
三、確保版本對應
確保你的Unity版本和pip上安裝的版本相匹配。你可以在 https://github.com/Unity-Technologies/ml-agents/releases/tag/release_20 的Releases部分查看對應關(guān)系。例如我用的ML-Agents Release 4版本配對如下:
com.unity.ml-agents (C#):v1.2.0
mlagents (Python):v0.18.0
mlagents-envs (Python):v0.18.0
gym-unity (Python):v0.18.0
Communicator (C#/Python):v1.0.0
檢查版本的步驟如下:
1.檢查Unity項目的ML-Agents版本:在Unity編輯器中,打開Window > Package Manager。在打開的窗口中,找到ML-Agents包并查看其版本號。
2.檢查訓練環(huán)境的ML-Agents版本:在命令行終端中,運行以下命令:
pip show mlagents
3.更新命令行工具MLAgents和mlagents-envs到對應版本,命令如下:
pip install --upgrade mlagents==0.18.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install --upgrade mlagents-envs==0.18.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
參考資料:https://zhuanlan.zhihu.com/p/82617680