Following instructions here
https://www.tensorflow.org/extend/adding_an_op

To try to rebuild this op
https://github.com/yaroslavvb/max_align_bytes_op

First I ran into issue with nsync headers, fixed by following
https://github.com/tensorflow/tensorflow/issues/12482#issuecomment-328829250

Then while trying to load the .so file I run into
tensorflow.python.framework.errors_impl.NotFoundError: ./max_align_bytes_op.so: undefined symbol: _ZTIN10tensorflow8OpKernelE

solution:
TF_LIB=$(python -c ‘import tensorflow as tf; print(tf.sysconfig.get_lib())’)

And add to the compiler flags:
-L$TF_LIB -ltensorflow_framework

2018-04-24 13:38:36.842975: I
tensorflow/core/platform/cpu_feature_guard.cc:137] Your CPU supports
instructions that this TensorFlow binary was not compiled to use:
SSE4.1 SSE4.2 AVX AVX2 FMA 2018-04-24 13:38:37.166540: I
tensorflow/stream_executor/cuda/cuda_gpu_executor.cc:892] successful
NUMA node read from SysFS had negative value (-1), but there must be
at least one NUMA node, so returning NUMA node zero 2018-04-24
13:38:37.166776: I
tensorflow/core/common_runtime/gpu/gpu_device.cc:1030] Found device 0
with properties: name: GeForce GTX 1070 major: 6 minor: 1
memoryClockRate(GHz): 1.759 pciBusID: 0000:01:00.0 totalMemory:
7.92GiB freeMemory: 7.22GiB 2018-04-24 13:38:37.166791: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1120] Creating
TensorFlow device (/device:GPU:0) -> (device: 0, name: GeForce GTX
1070, pci bus id: 0000:01:00.0, compute capability: 6.1)
Tensor(“Placeholder:0”, shape=(?, ?, ?, 3), dtype=float32)
Tensor(“conv5_3/conv5_3:0”, shape=(?, ?, ?, 512), dtype=float32)
Tensor(“rpn_conv/3x3/rpn_conv/3x3:0”, shape=(?, ?, ?, 512),
dtype=float32) Tensor(“rpn_cls_score/rpn_cls_score:0”, shape=(?, ?, ?,
18), dtype=float32) Tensor(“rpn_cls_prob:0”, shape=(?, ?, ?, ?),
dtype=float32) Tensor(“rpn_cls_prob_reshape:0”, shape=(?, ?, ?, 18),
dtype=float32) Tensor(“rpn_bbox_pred/rpn_bbox_pred:0”, shape=(?, ?, ?,
36), dtype=float32) Tensor(“Placeholder_1:0”, shape=(?, 3),
dtype=float32) Tensor(“conv5_3/conv5_3:0”, shape=(?, ?, ?, 512),
dtype=float32) Tensor(“rois:0”, shape=(?, 5), dtype=float32)
[

Logo

瓜分20万奖金 获得内推名额 丰厚实物奖励 易参与易上手

更多推荐