"Check whether your GraphDef-interpreting binary is up to date with your GraphDef-generating binary."

"Check whether your GraphDef-interpreting binary is up to date with your GraphDef-generating binary."

我正在关注他们社区版的数据块教程:

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.).'

*我卡住了,请帮帮我:)

我在 tensorflow 1.4.0 上使用了它。

解决方案是将您的 tensorflow 降级到 1.4.0

如果有人遇到这个问题,那么执行 pip uninstall tensorflow-gpu 并再次安装它对我有用。多半是版本问题。我现在是 1.14。