安裝 TensorFlow for Ubuntu

如果您想要根據 TensorFlow 模型來對影像進行評分,則必須在 Edge 上準備 TensorFlow 環境。

程序

  1. 使用下列指令安裝 tensorflow-gpu 1.4.0。請確定在安裝期間未報告任何錯誤,而且所有相依關係套件的版本都是正確的。
    pip install tensorflow-gpu==1.4.0
  2. 在 Python 中匯入 TensorFlow 以驗證 TensorFlow 是否正確地安裝。如果沒有錯誤,則表示 TensorFlow 已順利地安裝。
    import tensorflow