TF-deformable-conv icon indicating copy to clipboard operation
TF-deformable-conv copied to clipboard

have you tried it in tensorflow1.4?

Open Zealoe opened this issue 7 years ago • 2 comments

I used it in tensorflow1.4, and met the bug, tensorflow.python.framework.errors_impl.NotFoundError: /home/master/Deform/lib/deform_conv.so: undefined symbol: _ZTIN10tensorflow8OpKernelE and my g++ is 4.9.4. nvcc_compile.sh is as below: nvcc -std=c++11 -arch=sm_37 -ccbin=/usr/bin/g++-4.9 -c -o deform_conv.cu.o deform_conv.cu.cc -I $TF_INC -D GOOGLE_CUDA=1 -x cu -Xcompiler -fPIC -L /usr/local/cuda-8.0/lib64/ --expt-relaxed-constexpr

g++_compile.sh is as below: g++-4.9 -std=c++11 -shared -o deform_conv.so deform_conv.cc deform_conv.cu.o -I $TF_INC -fPIC -lcudart -L $CUDA_HOME/lib64 -D GOOGLE_CUDA=1 -Wfatal-errors -I $CUDA_HOME/include -D_GLIBCXX_USE_CXX11_ABI=0

Zealoe avatar Dec 14 '17 08:12 Zealoe

No I didn't test it in tf 1.4. This issue suggests to downgrade to 1.3.

Zardinality avatar Dec 14 '17 10:12 Zardinality

According to this issue It might work when -L$TF_LIB -ltensorflow_framework are added to g+= flags. This flag actually baffles me.

Zardinality avatar Dec 14 '17 10:12 Zardinality