最近用到KD树,发现CUDA版本的特别快,记录下调用过程。
1.首先下载源码
Github: https://github.com/johnarobinson77/KdTreeGPU
2. 编译
make all
编译之后会生成一系列的.o文件
- 编译时,Makefile 中加入(不加后面有可能报错)
ALL_CCFLAGS += -Xcompiler -fPIC
3.生成动态链接库
g++ -fPIC -shared -std=c++11 -o lib_kdtree.so *.o -I /usr/local/cuda-8.0/include -lcudart -L /usr/local/cuda-8.0/lib64/ -O2 -D_GLIBCXX_USE_CXX11_ABI=0 -lgomp
生成了lib_kdtree.so文件
4.python 调用