準備 unity machine learning 的環境真不是普通的坑哪!!
縱使我在v0.3才開始摸, 各種環節還是各種坑…
其中對一般人最困擾的莫過於各種組件的"版本"!!
所以這篇先來解決確認版本的步驟.
目前這個時間點, tensorflow-gpu 會明確使用特定的 DLL, 所以版本要非常一致才行.
安裝教學中也有特別提到這段:
As of ML-Agents v0.3, only CUDA 8 and cuDNN 6 is supported.
- 目前找到一個比較可靠的方式是:
TensorFlow 的版本, 從 requirements.txt 找到. - TensorFlow-gpu 的版本, 可從 TensFlow 的 Linux 安裝教學頁面的 “The URL of the TensorFlow Python package" 區塊得知.
- CUDA & cuDNN 的版本取得方式有兩種:
a. 從 ml-agents 的 Windows 安裝教學頁面得知,
b. 從 TensFlow 的 Linux 安裝教學頁面 的 “NVIDIA requirements to run TensorFlow with GPU support" 區塊得知.
以追蹤 ml-agents v0.3 的版本資訊為例
- requirements.txt 裡面得知 tensorflow 的版本為 1.4.0
- 從 “NVIDIA requirements to run TensorFlow with GPU support" 區塊得知 CUDA v8.0, cuDNN v6.0.
- 從 “The URL of the TensorFlow Python package" 區塊得知 tensorflow-gpu 版本是 v1.4.0.
※GitHub 竟然不能搜尋一些特殊字元, 傻眼…
One comment