TensorFlow導入失敗

如果集群上安裝了不兼容的protobuf版本,TensorFlow將無法導入。

寫的kavya.parag

最後發布日期:2022年5月16日

問題

你有TensorFlow安裝在您的集群上。

當你嚐試導入TensorFlow,它失敗了無效的語法導入錯誤

導致

版本protobuf的版本不兼容TensorFlow

解決方案

使用集群範圍的初始化腳本進行安裝TensorFlow有匹配版本的NumPy而且protobuf

  1. 創建初始化腳本。
    %python dbutils.fs.put("/databricks//install-tensorflow.sh",""" #!/bin/bash set -e /databricks/python/bin/python -V /databricks/python/bin/pip install tensorflow protobuf==3.17.3 numpy==1.15.0 """, True)
  2. 安裝剛才創建的集群範圍的初始化腳本(AWS|Azure|GCP).
    您需要腳本所在位置的完整路徑(dbfs: /磚/ < init-script-folder > / install-tensorflow.sh).
  3. 安裝init腳本後重新啟動集群。
刪除

信息

的所有現有版本NumPy在集群上安裝初始化腳本之前。


這篇文章有用嗎?