light_head_rcnn
light_head_rcnn copied to clipboard
light_head_rcnn/lib/lib_kernel/lib_psalign_pooling/psalign_pooling.so: cannot open shared object file: No such file or directory
When run the command "python3 train.py -d 0-7"
it shows error as following:
Traceback (most recent call last):
File "train.py", line 12, in
- I have compiled lib folder successfully
- I checked the folder "light_head_rcnn/lib/lib_kernel/lib_psalign_pooling/", and there is no file named "psaligh_pooling.so". I am confusing that is there anything wrong when I compiled the lib
@hang0522 "python3 train.py -d 0-7" 0-7 show your gpu id so if u are using one gpu you need to specify it by only 0 "python3 train.py -d 0"
I am having issue when loading the session from .meta file `
Traceback (most recent call last):
File "freeze_graph.py", line 66, in
Since PSROIPool is implemented in .so as a layer is it possible to freeze_graph ?
vi mvpose/backend/light_head_rcnn/lib/lib_kernel/lib_psalign_pooling/make.sh
TF_INC=$(python3 -c 'import tensorflow as tf; print(tf.sysconfig.get_include())')
TF_LIB=$(python3 -c 'import tensorflow as tf; print(tf.sysconfig.get_lib())')
CUDA_PATH=/usr/local/cuda-9.0/ # jing
nvcc -std=c++11 -c -o psroi_pooling_op.cu.o psroi_pooling_op_gpu.cu.cc
-I /usr/local \ # jing
-I $TF_INC -D GOOGLE_CUDA=1 -x cu -Xcompiler -fPIC -arch=sm_52 -I$TF_INC/external/nsync/public --expt-relaxed-constexpr
g++ -std=c++11 -shared -D_GLIBCXX_USE_CXX11_ABI=0 -o psroi_pooling.so psroi_pooling_op.cc
psroi_pooling_op.cu.o -I $TF_INC -fPIC -lcudart -L $CUDA_PATH/lib64 -L$TF_LIB -ltensorflow_framework -I$TF_INC/external/nsync/public