＆＃34;检查GraphDef解释二进制文件是否与生成GraphDef的二进制文件保持同步。＆＃34;

Question

我正在关注社区版的数据库教程：

https://docs.databricks.com/applications/deep-learning/deep-learning-pipelines.html#install-deep-learning-pipelines

附加图书馆：

  

spark-deep-learning，tensorflow，keras，h5py，tensorframes

运行此单元格时：

from pyspark.ml.classification import LogisticRegression
from pyspark.ml import Pipeline
from sparkdl import DeepImageFeaturizer

featurizer = DeepImageFeaturizer(inputCol="image", outputCol="features", modelName="InceptionV3")
lr = LogisticRegression(maxIter=20, regParam=0.05, elasticNetParam=0.3, labelCol="label")
p = Pipeline(stages=[featurizer, lr])

p_model = p.fit(train_df)

我收到此错误：

IllegalArgumentException: u'NodeDef mentions attr \'dilations\' not in Op<name=Conv2D; signature=input:T, filter:T -> output:T; attr=T:type,allowed=[DT_HALF, DT_FLOAT]; attr=strides:list(int); attr=use_cudnn_on_gpu:bool,default=true; attr=padding:string,allowed=["SAME", "VALID"]; attr=data_format:string,default="NHWC",allowed=["NHWC", "NCHW"]>; NodeDef: given/conv2d_95/convolution = Conv2D[T=DT_FLOAT, data_format="NHWC", dilations=[1, 1, 1, 1], padding="VALID", strides=[1, 2, 2, 1], use_cudnn_on_gpu=true](given/sub_1, given/conv2d_95/kernel/read). (Check whether your GraphDef-interpreting binary is up to date with your GraphDef-generating binary.).'

*我被困了请帮帮我:)。

Answer 1

我使用了tensorflow 1.4.0。

Answer 2

解决方案正在将张量流降级到1.4.0

Answer 3

如果有人遇到此问题，请执行pip卸载tensorflow-gpu并再次安装它对我有用。最有可能是版本问题。我现在是1.14。