[CUDA实战] 第一个样例程序

这两天开始看《GPU高性能编程CUDA实战》这本书,学到的东西稍稍在博客做个记录。


使用cuda c编程的话,当然首先需要有个支持cuda的gpu,市面上很多gpu都已经支持cuda,之后需要为gpu安装cuda驱动程序,我使用的是ubuntu系统,安装方法在这篇博客 cuda7.5 安装 中有提到,可作参考,当然也要安装个gcc、g++(自己习惯用c++)编译器,这个就不说了。


根据书上的内容,写了第一个cuda程序,代码文件名为cuda.cu,代码如下:

__global__ void add(int a, int b, int *c)//kernel函数,在gpu上运行。
{
    *c = a + b;
}

int main()
{
    int c;
    int *dev_c;
    cudaMalloc((void**)&dev_c, sizeof(int));//分配gpu的内存,第一个参数指向新分配内存的地址,第二个参数是分配内存的大小。
    add<<<1,1>>>(2, 7, dev_c);//调用kernel函数,<<<1,1>>>指gpu启动1个线程块,每个线程块中有1个线程。
    cudaMemcpy(&c, dev_c, sizeof(int), cudaMemcpyDeviceToHost);//将gpu上的数据复制到主机上,
	                                                           //即从dev_c指向的存储区域中将sizeof(int)个字节复制到&c指向的存储区域。
    cout << "2 + 7 = " << c << endl;
    cudaFree(dev_c);//释放cudaMalloc分配的内存。
    return 0;
}

然后进行编译:nvcc cuda.cu -o cuda
生成名为cuda的可执行文件
之后执行:./cuda
执行结果:2 +  7 =  9

你可能感兴趣的:(cuda)