Cuda dll封装

  1. 新建Cuda工程 –Cuda 7.5 Runtime
  2. 项目-右键属性-配置属性 修改配置类型为dll。
  3. 添加一个头文件和一个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库。

你可能感兴趣的:(Cuda dll封装)