(2)查询CUDA设备

来源:互联网 发布:sas软件哪些人要学 编辑:程序博客网 时间:2024/05/17 04:09

调用CUDA提供API查询当前计算机支持CUDA 的设备信息:

(打印出部分信息)

 

#include "stdafx.h"
#include "cuda_runtime_api.h"

int main(int argc, _TCHAR* argv[])
{

 //get Devices count 
 int nDevCount = -1;
 cudaError_t nErr;

 nErr = cudaGetDeviceCount(&nDevCount);

 if (cudaSuccess != nErr)
 {
  printf("Get devices count failed !\n");
  return -1;
 }
 if (nDevCount <= 0)
 {
  printf("You have no graphics card support CUDA !\n");
 }
 else
 {
  printf("You have %d graphics card support CUDA !\n", nDevCount);
 }

 //Get devices info
 cudaDeviceProp pProp;
 nErr = cudaGetDeviceProperties(&pProp, 0);
 if (cudaSuccess != nErr)
 {
  printf("Get devices info failed !\n");
  return -1;
 }
 printf("Name: %s\n", pProp.name);
 printf("totalGlobalMemory: %lld\n", pProp.totalGlobalMem);
 printf("totalConstMemory :%lld\n", pProp.totalConstMem);
 printf("clockRate: %d\n", pProp.clockRate);
 printf("multiProcessorCount :%d\n", pProp.multiProcessorCount);
 printf("major:%d\n", pProp.major);
 printf("minor :%d\n", pProp.minor);
 printf("maxThreadsPerBlock :%d\n", pProp.maxThreadsPerBlock);

 int i = 0;
 scanf("%d", &i);
 
 return 0;
}

运行结果:

原创粉丝点击