Skip to content

CuDNN

Simple example

CUDA코드는 아래와 같다.

// main.cu
#include <stdio.h>
#include <cudnn.h>

int main()
{
    cudnnHandle_t hCudNN = NULL;
    cudnnStatus_t status = cudnnCreate(&hCudNN);

    if (status == CUDNN_STATUS_SUCCESS) {
        printf("cuDNN create.\n");

        if (hCudNN != NULL) {
            printf("cuDNN destroy.\n");
            cudnnDestroy(hCudNN);
        }
    } else {
        printf("CUDA Status: error.\n");
        return 1;
    }

    return 0;
} 

아래와 같이 컴파일 하면 된다.

nvcc -L/usr/local/cudnn/lib64 -lcudnn main.cu

See also

Favorite site