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;
}
아래와 같이 컴파일 하면 된다.
See also
Favorite site
- cuDNN examples
- [추천] Deep Learning Courses
- Github: cudnn-training (A CUDNN minimal deep learning training code sample using LeNet)