hdrnet_legacy
hdrnet_legacy copied to clipboard
Problem running the train script
I am trying to train a model after making the hdrnet_ops, but I am getting the following error:
line 60, in load_op_library
lib_handle = py_tf.TF_LoadLibrary(library_filename)
tensorflow.python.framework.errors_impl.NotFoundError: hdrnet-master/hdrnet/lib/hdrnet_ops.so: undefined symbol: _ZN10tensorflow8OpKernelD2Ev
I am using the following setting: python 2.7.15, tensorflow 1.12, gcc 5.5.0 and cuda 10.
I appreciate getting your help on this.
you can add ltensorflow_framework to nvcc link flags to solve this problem
Also need to add the following line in the makefile and use tf_lib in the link flags.
TF_LIB ?= python -c 'import tensorflow as tf; print(tf.sysconfig.get_lib())'
Hi @cfanyyx @seasonpplp how can i get the libtensorflow_framework.so correctly? I installed tensorlow-gpu-1.1.0 according the requiremet.txt, but there is no libtensorflow_framework.so is the tensorflow root path (/home/xxxx/venv2.7/local/lib/python2.7/site-packages/tensorflow/core), so I download the source code (branch r1.1) and build using bazel commands, but the so i got is libtensorflow_cc.so and there was no libtensorflow_framework.so in the bazel-bin directory. would you like to tell me the building steps, thanks. I wonder if there are mistakes in versions I selected