如何创建 python 中使用的 cuda 共享库

how to create a cuda shared library used in python

我创建了一个用于 python 的 C 共享库,当我测试它时,它工作得很好。 我这样编译:

 gcc -shared -std=c99 -I/usr/include/python2.6 -fPIC -lpython2.6 -opymod.so pymod.c

我有一个 cuda 源代码 "pymod.cu" 和 "pymod.c" 一样,我真的不知道如何用 nvcc 编译它。

提前致谢。

您需要先使用 nvcc 编译,然后使用 gcc 构建共享库:

nvcc -c test.cu -o test.o -Xcompiler -fPIC 
gcc -shared -fPIC -o libtest.so test.o