TensorFlow無法導入

TensorFlow未能導入如果你有一個不兼容的版本的protobuf安裝在您的集群。

寫的kavya.parag

去年發表在:2022年5月16日

問題

你有TensorFlow安裝在您的集群。

當你試圖導入TensorFlow,它與一個失敗無效的語法或導入錯誤。

導致

的版本protobuf安裝在您的集群和你的版本不兼容TensorFlow。

解決方案

使用一個集群級init腳本安裝TensorFlow與匹配的版本NumPy和protobuf。

  1. 創建初始化腳本。
    % python dbutils.fs.put(“/磚/ < init-script-folder > / install-tensorflow。sh”、“”“# !/bin/bash設置- e /磚/ python / bin / python - v /磚/ python / bin / pip安裝tensorflow protobuf = = 3.17.3 numpy = = 1.15.0”“”,真的)
  2. 安裝您剛剛創建的init腳本作為一個集群級init腳本(AWS|Azure|GCP)。
    你需要的完整路徑的位置腳本(dbfs: /磚/ < init-script-folder > / install-tensorflow.sh)。
  3. 安裝後重新啟動集群init腳本。
刪除

信息

卸載所有現有版本的NumPy在安裝之前在您的集群init腳本。