环境
系统:Ubuntu18.04.4
tensorflow:2.2.0
python:3.7.6
cuda:10.2.89
cudnn:7.6.5
一、编译
1.1编译tf_custom_ops
作者的环境是tensorflow1.x,直接使用作者提供的compile_op.sh脚本编译会提示找不到ltensorflow_framework
问题出在编译命令行中的 -ltensorflow_framework参数,同时作者还指出如果使用的是Ubuntu18.04的系统需要把参数-D_GLIBCXX_USE_CXX11_ABI=0去掉,但是我没有去也可以正常执行。
下面说说怎么修改,首先查看自己的环境下的链接和编译参数
import tensorflow as tf
print