#include "cuda_runtime.h" #include "device_launch_parameters.h" #include<iostream> using namespace std; __global__ void hello() { cout << "hello word GPU\n"; } int main(int argc, char** argv) { cout << "hello word CPU\n"; hello << <1, 10 >> > (); cudaDeviceReset(); return 0; }
时间: 2024-04-28 12:21:52 浏览: 154
使用include <iostream>时一定要加using namespace std
这段代码是一个简单的 CUDA 程序,它使用了 NVIDIA 的 CUDA 并行计算平台。CUDA 可以让程序员通过编写 CUDA C/C++ 代码来利用 NVIDIA GPU 的并行计算能力,从而加速程序的运行。
具体来说,这段代码定义了一个名为 `hello` 的 CUDA Kernel 函数,用于在 GPU 上输出字符串 "hello word GPU"。然后在主函数中,我们先输出字符串 "hello word CPU",然后使用 CUDA 的并行计算方式,启动一个由 10 个线程组成的 CUDA 线程块来执行 `hello` 函数。最后使用 `cudaDeviceReset()` 来清除并释放所有 CUDA 设备上的资源。
需要注意的是,由于 CUDA 只能在支持 CUDA 的 NVIDIA GPU 上运行,因此在运行这段代码之前需要确保您的计算机上已经安装了 NVIDIA 的 GPU 驱动程序和 CUDA 工具包。
阅读全文