【CUDA-2】第一個CUDA代碼例子——hello world

在上一篇文章中【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運算能力的差別。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章