在上一篇文章中【CUDA-1】centos7安裝CUDA,記錄了在centos7安裝cuda的過程,這一篇,我們首先用一個最簡單的示例代碼來感受一下,CUDA編程的基本風格。
少廢話,上代碼:
// 001_hello.cu
#include <stdio.h>
#include <unistd.h>
// __global__ 修飾符,將告訴編譯器,函數在設備(GPU)上運行而不是在主機(CPU)上運行
__global__ void kernel(void)
{
printf("Hello world!\n");
}
int main(void)
{
while(1)
{
kernel<<<1,1>>>();
sleep(1);
}
return 0;
}
代碼怎麼編譯呢?超級簡單,只需要將gcc 換成nvcc就好了。
nvcc 001_hello.cu -o e_002
爲了只管的看見GPU的佔用,上輸代碼中使用了循環一直執行代碼 kernel<<<1,1>>>();
可以發現,幾乎所有的語法和C語言一模一樣。
執行結果:
體驗了CUDA基本語法之後,在下一篇中將會講解如何獲取GPU的屬性,以及運行以個並行計算的例子,並比較CPU和GPU運算能力的差別。