問題
你有TensorFlow安裝在您的集群上。
當你嚐試導入TensorFlow,它失敗了無效的語法或導入錯誤.
導致
版本protobuf的版本不兼容TensorFlow.
解決方案
使用集群範圍的初始化腳本進行安裝TensorFlow有匹配版本的NumPy而且protobuf.
- 創建初始化腳本。
%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) - 安裝剛才創建的集群範圍的初始化腳本(AWS|Azure|GCP).
您需要腳本所在位置的完整路徑(dbfs: /磚/ < init-script-folder > / install-tensorflow.sh). - 安裝init腳本後重新啟動集群。