[英]Calling C++ CUDA code from python
我想從python調用一個用CUDA(C ++)編寫的函數,並將numpy數組作為輸入傳遞給它,並從這個函數中獲取輸出數組。 這可能嗎? 這是這個問題的唯一目標。
ps:據我所知,如果要將這樣的函數轉換為正確的程序並編譯成可執行文件,那么可以從python中調用它作為shell命令。 但是,如何在不使用文件的情況下傳遞和獲取數組作為輸入和輸出?
PyCuda由nvidia提供。 它也很容易使用。 看看它https://developer.nvidia.com/pycuda
我想你想要這樣的東西:
import pycuda.autoinit
import pycuda.driver as drv
import numpy
from pycuda.compiler import SourceModule
mod = SourceModule("""
__global__ void multiply_them(float *dest, float *a, float *b)
{
const int i = threadIdx.x;
dest[i] = a[i] * b[i];
}
""")
multiply_them = mod.get_function("multiply_them")
a = numpy.random.randn(400).astype(numpy.float32)
b = numpy.random.randn(400).astype(numpy.float32)
dest = numpy.zeros_like(a)
multiply_them(
drv.Out(dest), drv.In(a), drv.In(b),
block=(400,1,1), grid=(1,1))
print dest-a*b
有一些庫可能對這類任務有所幫助,但PyCuda似乎與你所描述的最接近。
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.