Tensorflow Extended(TFX)代码在Google Colab上运行良好,但在本地计算机上运行时会引发错误

时间:2020-05-27 12:05:47

标签: tensorflow tfx

我使用Tensorflow Extended(TFX)使用以下代码

from tfx.utils.dsl_utils import csv_input
from tfx.components.example_gen.csv_example_gen.component import CsvExampleGen

examples = csv_input(os.path.join(base_dir, 'data/simple'))
example_gen = CsvExampleGen(input=examples);

当我在Google Colab上执行此代码时,它可以正常工作。但是,当我运行此

回溯(最近通话最近):

中的文件“ tfx_sample.py”,第4行 从tfx.components.example_gen.csv_example_gen.component导入CsvExampleGen

中的文件“ /Users/sv/tfx_env/lib/python3.7/site-packages/tfx/components/init.py”,第20行 从tfx.components.bulk_inferrer.component导入BulkInferrer
文件“ /Users/sv/tfx_env/lib/python3.7/site-packages/tfx/components/bulk_inferrer/component.py”,
第24行,
从tfx.components.base导入base_component
文件“ /Users/sv/tfx_env/lib/python3.7/site-packages/tfx/components/base/base_component.py”,第28行,
从tfx.components.base导入base_driver
文件“ /Users/sv/tfx_env/lib/python3.7/site-packages/tfx/components/base/base_driver.py”,第28行,
从tfx.orchestration导入元数据
文件“ /Users/sv/tfx_env/lib/python3.7/site-packages/tfx/orchestration/metadata.py”,第36行,在 从ml_metadata.metadata_store导入元数据_store
在第15行的“ /usr/local/lib/python3.7/site-packages/ml_metadata/metadata_store/init.py”文件中 从ml_metadata.metadata_store.metadata_store导入MetadataStore
文件“ /usr/local/lib/python3.7/site-packages/ml_metadata/metadata_store/metadata_store.py”,第32行,在 从ml_metadata.metadata_store导入pywrap_tf_metadata_store_serialized为metas_store_serialized
第28行中的文件“ /usr/local/lib/python3.7/site-packages/ml_metadata/metadata_store/pywrap_tf_metadata_store_serialized.py” _pywrap_tf_metadata_store_serialized = swig_import_helper()
swig_import_helper中的第24行,文件“ /usr/local/lib/python3.7/site-packages/ml_metadata/metadata_store/pywrap_tf_metadata_store_serialized.py” _mod = imp.load_module('_ pywrap_tf_metadata_store_serialized',fp,路径名,描述)
在load_module中的文件“ /usr/local/Cellar/python/3.7.7/Frameworks/Python.framework/Versions/3.7/lib/python3.7/imp.py”,第242行 返回load_dynamic(名称,文件名,文件)
在load_dynamic中的文件“ /usr/local/Cellar/python/3.7.7/Frameworks/Python.framework/Versions/3.7/lib/python3.7/imp.py”,第342行 返回_load(spec)
ImportError:dlopen(/usr/local/lib/python3.7/site-packages/ml_metadata/metadata_store/_pywrap_tf_metadata_store_serialized.so,2):找不到符号:____chkstk_darwin
引用自:/usr/local/lib/python3.7/site-packages/ml_metadata/metadata_store/_pywrap_tf_metadata_store_serialized.so(为Mac OS X 10.15构建)。
预期位于:/usr/lib/libSystem.B.dylib 在/usr/local/lib/python3.7/site-packages/ml_metadata/metadata_store/_pywrap_tf_metadata_store_serialized.so

这是已安装软件包的版本详细信息

apache-beam 2.20.0
张量板1.15.0
tensorboard-plugin-wit 1.6.0.post3
tensorflow 1.15.0
tensorflow-data-validation 0.23.0.dev0
tensorflow-estimator 1.15.1
tensorflow-元数据0.22.0
张量流模型分析0.21.5
tensorflow-serving-api 2.1.0
tensorflow转换0.22.0
tfx 0.21.4
tfx-bsl 0.22.0

1 个答案:

答案 0 :(得分:0)

请使用可用于 tfx 的最新版本 (0.29.0)。 解决问题的工作代码:-

import os
import tfx
from tfx.utils.dsl_utils import external_input
from tfx.components.example_gen.csv_example_gen.component import CsvExampleGen
import tempfile
import urllib

_data_root = tempfile.mkdtemp(prefix='tfx-data')
DATA_PATH = 'https://raw.githubusercontent.com/tensorflow/tfx/master/tfx/examples/chicago_taxi_pipeline/data/simple/data.csv'
_data_filepath = os.path.join(_data_root, "data.csv")
urllib.request.urlretrieve(DATA_PATH, _data_filepath)

example_gen = CsvExampleGen(input_base=(_data_filepath))