【python】python 调用CUDA动态链接库

本文介绍了如何使用Python调用CUDA动态链接库来加速KD树的运算。首先从Github下载源码,然后进行编译生成.o文件。接着通过Makefile添加必要设置,编译得到lib_kdtree.so动态链接库。最后,演示了在Python中仅能调用main接口来调用该库,其他接口的调用留待后续研究。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

最近用到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 调用

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值