主要记录一下训练PointSIFT的训练过程
https://github.com/MVIG-SJTU/pointSIFT.git
先给出代码地址,git到本地
根据git给出的流程训练即可
1.安装环境(不在赘述)
可以按照官网的版本。我这里是
g++ --version查看g++版本.我的是7.5
2. 按照给出的方法将四个文件中的sh文件编译,
我的配置文件是
#/bin/bash
/usr/local/cuda-10.0/bin/nvcc tf_sampling_g.cu -o tf_sampling_g.cu.o -c -O2 -DGOOGLE_CUDA=1 -x cu -Xcompiler -fPIC
# TF1.2
#g++ -std=c++11 tf_sampling.cpp tf_sampling_g.cu.o -o tf_sampling_so.so -shared -fPIC -I /usr/local/lib/python2.7/dist-packages/tensorflow/include -I /usr/local/cuda-8.0/include -lcudart -L /usr/local/cuda-8.0/lib64/ -O2 -D_GLIBCXX_USE_CXX11_ABI=0
# TF1.4
g++ -std=c++11 tf_sampling.cpp tf_sampling_g.cu.o -o tf_sampling_so.so -shared -fPIC -I /home/xuekelou420/.conda/envs/fpoint/lib/python3.6/site-packages/tensorflow/include -I /usr/local/cuda-10.0/include -I /home/xuekelou420/.conda/envs/fpoint/lib/python3.6/site-packages/tensorflow/include/external/nsync/public -lcudart -L /usr/local/cuda-10.0/lib64/ -L/home/xuekelou420/.conda/envs/fpoint/lib/python3.6/site-packages/tensorflow -ltensorflow_framework -O2
g++版本太高,将D_GLIBCXX_USE_CXX11_ABI=0删除或者置位1应该也可以。5.4以下不用删除
3.训练
两张显卡特斯拉先练起来了