由于一个系统可能有多个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));
}