[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