first CUDA

来源：互联网发布：电力大数据应用编辑：程序博客网时间：2024/05/16 19:39

CUDA 的初始化

#include <stdio.h>
#include <cuda_runtime.h>//要使用 runtime API 的时候，需要 include cuda_runtime.h

bool InitCUDA()//InitCUDA 函式，会呼叫 runtime API 中，有关初始化 CUDA 的功能：
{
    int count;

    cudaGetDeviceCount(&count);
    if(count == 0) {
        fprintf(stderr, "There is no device./n");
        return false;
    }

    int i;
    for(i = 0; i < count; i++) {
        cudaDeviceProp prop;
        if(cudaGetDeviceProperties(&prop, i) == cudaSuccess)

{
            if(prop.major >= 1) {
              break;
            }
        }
    }

    if(i == count) {
        fprintf(stderr, "There is no device supporting CUDA 1.x./n");
        return false;
    }

    cudaSetDevice(i);

    return true;
}

///////////////////////////////

cudaGetDeviceCount 函式，取得支持 CUDA 的装置的数目.

确定系统上是否有支持 CUDA 的装置，需要对每个 device 呼叫 cudaGetDeviceProperties 函式，取得装置的各项数据，并判断装置支持的 CUDA 版本（prop.major 和 prop.minor 分别代表装置支持的版本号码，例如 1.0 则 prop.major 为 1 而 prop.minor 为 0）.除了装置支持的 CUDA 版本之外，还有装置的名称、内存的大小、最大的 thread 数目、执行单元的频率等等。

在找到支持 CUDA 1.0 以上的装置之后，就可以呼叫 cudaSetDevice 函式，把它设为目前要使用的装置.

最后是 main 函式。在 main 函式中我们直接呼叫刚才的 InitCUDA 函式，并显示适当的讯息：

int main()
{
    if(!InitCUDA()) {
        return 0;
    }

    printf("CUDA initialized./n");

    return 0;
}

0 0