20211112：CUDA编程-获取GPU型号和属性

本文主要是介绍20211112：CUDA编程-获取GPU型号和属性，对大家解决编程问题具有一定的参考价值，需要的程序猿们随着小编来一起学习吧！

//获取GPU名字代码：
int cudaGetGpuName(char *name[], int *Num) {
  int i;
  int Count = 0;

  cudaError_t error_id = cudaGetDeviceCount(&Count);
  if (error_id != cudaSuccess) {
    APP_ERR("cuda get device count failed, ret:%d, %s", (int)error_id, cudaGetErrorString(error_id));
    return -1;
  }
  if (Count > 16) {
    APP_WARRING("too large gpu num : %d", deviceCount);
    Count = 16;
  }
  // printf("cudaDeviceProp size:%d\n", sizeof(cudaDeviceProp));
  // char datastr[1024];
  for (i = 0; i < Count; i++) {
    cudaSetDevice(i);
    // prevent device version problem, not use struct cudaDeviceProp
    cudaDeviceProp Prop;
    cudaGetDeviceProperties(&Prop, i);
    // printf("\nDevice %d: \"%s\"\n", i, datastr);
    name[i] = (char *)malloc(256);
    memset(name[i], 0, 256);
    if (name[i] != NULL) {
      strncpy(name[i], Prop.name, 256);
    } else {
      APP_ERR("malloc failed");
      return -1;
    }
  }
  *Num = Count;
  return 0;
}

这篇关于20211112：CUDA编程-获取GPU型号和属性的文章就介绍到这儿，希望我们推荐的文章对大家有所帮助，也希望大家多多支持为之网！