Unity – ml-agents 環境準備2 – 安裝

一直找不到時間弄個圖文版, 拖到現在就直接用文字帶過囉~ 哈哈!!

延續上一篇確認版本之後就可以開始安裝了~

這裡假設你是要用到 Nvidia 的 gpu 運算.

以下以 ml-agents v3.0 on win10 的版本為範例, 所需的版本是 TensorFlow 1.4.0 & TensorFlow-gpu 1.4.0 & CUDA 8 & cuDNN 6(for CUDA 8).

大綱

  1. 安裝 CUDA 8 & cuDNN 6(for CUDA 8).
  2. 安裝 Anaconda(理論上版本不限).
  3. 在 Anaconda 產生新"環境".
  4. 在該環境中安裝所需元件(TensorFlow, TensorFlow-gpu, …等等).
  5. 測試環境

細節

1 & 2的步驟都只是一般軟體安裝就不贅述, 請直接參考官方文件 or Unity Taiwan 的攻略
※版本很重要! 版本很重要! 版本很重要!

開啟 Anaconda Prompt(建議直接用管理者權限開啟).

產生"環境"並安裝python 3.6版
(base) C:\Users\xxx>conda create -n ml-agents python=3.6

啟用"環境", 從這開始 command line 的最前面會變成 (ml-agents) 開頭, 這是你處於名為 “ml-agents" 的 “環境" 底下的指標.
(base) C:\Users\xxx>conda activate ml-agents

透過 requirements.txt 安裝所需元件, 這裡假設把 ml-agents github 的專案放在 “D:\ml-agents".
pip install D:\ml-agents\python

另外手動安裝 tensorflow-gpu 1.4.0 版
(ml-agents) C:\Users\xxx>pip install tensorflow-gpu==1.4.0

到這裡其實就安裝完成了!

接著要要用來驗證環境是否正確的, 嘗試在 python 裡面 import tensorflow.

啟動 python, 如果有重開, 記得先用 conda activate 切換到正確的環境.
(ml-agents) C:\WINDOWS\system32>python

import TensFlow
>>> import tensorflow as tf

如果沒有跳出任何訊息就是好消息!!
反之出現一大串 Excpetion 之類的訊息就是有過程失誤了~

goodluck!!

發表迴響

在下方填入你的資料或按右方圖示以社群網站登入:

WordPress.com 標誌

您的留言將使用 WordPress.com 帳號。 登出 /  變更 )

Google photo

您的留言將使用 Google 帳號。 登出 /  變更 )

Twitter picture

您的留言將使用 Twitter 帳號。 登出 /  變更 )

Facebook照片

您的留言將使用 Facebook 帳號。 登出 /  變更 )

連結到 %s

This site uses Akismet to reduce spam. Learn how your comment data is processed.