GPU设备管理

  由于一个系统可能有多个GPU,因此在进行 cuda编程中获取GPU设备信息并选取合适的GPU用于运行程序是很好的习惯。以下代码实现了这两个目的。

int main(int argc,char **argv)
{
// set up device
     int dev = 0;
     cudaDeviceProp deviceProp;
     CHECK(cudaGetDeviceProperties(&deviceProp, dev));
     printf("%s at ", argv[0]);
     printf("device %d: %s ", dev, deviceProp.name);
     CHECK(cudaSetDevice(dev));
}

你可能感兴趣的:(CUDA)