TensorFlow 1.4.1模型C++推理报错:未注册HashTableV2 OpKernel
解决TensorFlow 1.4.1 C++推理时"HashTableV2" OpKernel未注册的问题
你遇到的问题核心是手动编译TensorFlow静态库时,没有包含HashTableV2对应的OpKernel实现,再加上在MacOS上误用了Linux的编译脚本,进一步放大了兼容性问题。下面是针对性的解决步骤:
1. 替换为MacOS专属编译脚本
你当前在MacOS上运行build_all_linux.sh,这是为Linux平台设计的编译脚本,会导致平台相关的依赖和OpKernel不兼容。请切换到MacOS的编译脚本:
cd tensorflow/contrib/makefile ./build_all_macos.sh
2. 添加HashTableV2相关的OpKernel源文件
除了你已经添加的lookup_table_op.cc和initializable_lookup_table.cc,还需要把HashTableV2的核心实现文件加入编译列表:
- 在
tensorflow/contrib/makefile/Makefile的SOURCES部分,添加以下文件路径:
(注:TensorFlow 1.4.1中tensorflow/core/kernels/hash_table_ops.cc tensorflow/core/kernels/hash_table_v2_ops.ccHashTableV2的注册逻辑和核心实现就在这两个文件里,之前的编译未包含它们,才会出现OpKernel未注册的报错)
3. 补全Lookup表相关的依赖文件
为了避免后续出现其他Lookup类Op的问题,建议在Makefile中补充以下依赖文件:
tensorflow/core/kernels/lookup_table_init_op.cctensorflow/core/kernels/string_to_hash_bucket_op.cc
4. 清理旧编译产物后重新编译
为了避免缓存导致的编译残留问题,先清理之前的编译结果,再重新执行编译:
make clean -C tensorflow/contrib/makefile ./build_all_macos.sh
5. 验证版本一致性
确保训练模型时使用的TensorFlow版本(1.4.1)和C++编译的静态库版本完全一致,避免版本间的Op定义差异导致的兼容性问题。
完成以上步骤后,重新链接你的C++推理程序,应该就能解决"No OpKernel was registered to support Op 'HashTableV2'"的报错了。
内容的提问来源于stack exchange,提问作者Yan
相关产品推荐
相关产品推荐

