Cuda dll封装

新建Cuda工程 –Cuda 7.5 Runtime

项目-右键属性-配置属性 修改配置类型为dll。

添加一个头文件和一个cpp文件。

test.h test.cpp

test.h :

#ifndef  CUDA_TEST_H
#define CUDA_TEST_H
__declspec(dllexport) int GetGPUCount(void);
#endif

test.cpp

#include "test.h"
#include <cuda_runtime.h>
#pragma comment(lib,"cuda_static.lib")

int GetGPUCount(void){
int deviceCount = 0;
cudaError_t error_id = cudaGetDeviceCount(&deviceCount);
if(error_id != cudaSuccess){
printf("cudaGetDeviceCount returned %d\n->%s\n",(int)error_id,cudaGetErrorString(error_id));
return 0;
}
return deviceCount;
}

包含目录啥的都不需要配置。生成即可得到dll库。