[CUDA]GPU设备属性的查询和使用

要获得GPU设备的数量，在cuda中是通过cudaGetDeviceCount(int*)来实现的。

对于每个GPU设备。cuda使用cudaDeviceProp类型来保存设备信息。cudaDeviceProp的详细内容如下

 
 我们通过cudaGetDeviceProperties(cudaDeviceProp* prop,int i)来获取第i块设备属性。具体参考代码如下

#include<cuda_runtime.h>
#include<windows.h>
#include<iostream>
using namespace std;

int main(){
	int count;
	cudaDeviceProp prop;
	cudaGetDeviceCount(&count);
	cout <<"当前计算机包含GPU数为"<< count << endl;
	for (int i = 0; i < count; i++){
		cudaGetDeviceProperties(&prop, i);
		cout << "当前访问的是第" << i << "块GPU属性" << endl;
		cout << "当前设备名字为" << prop.name << endl;
		cout << "GPU全局内存总量为" << prop.totalGlobalMem << endl;
		cout << "单个线程块中包含的线程数最多为" << prop.maxThreadsPerBlock << endl;

	}
	getchar();
	return 0;
}

在cuda中一般使用cudaChooseDevice()来选取符合标准的硬件cudaChooseDevice()返回距离给定硬件标准值最近的硬件编号。我们把这个编号传递给cudaSetDevice()，然后所有的设备操作都在这个设备上进行。

 

#include<cuda_runtime.h>
#include<windows.h>
#include<iostream>
using namespace std;

int main(){
	int count;
	cudaDeviceProp prop;
	cudaGetDeviceCount(&count);
	cout <<"当前计算机包含GPU数为"<< count << endl;
	for (int i = 0; i < count; i++){
		cudaGetDeviceProperties(&prop, i);
		cout << "当前访问的是第" << i << "块GPU属性" << endl;
		cout << "当前设备名字为" << prop.name << endl;
		cout << "GPU全局内存总量为" << prop.totalGlobalMem << endl;
		cout << "单个线程块中包含的线程数最多为" << prop.maxThreadsPerBlock << endl;

	}

	int dev;
	memset(&prop, 0, sizeof(prop));
	prop.major = 1;
	prop.minor = 3;
	cudaChooseDevice(&dev, &prop);
	cout << "距离设备版本1.3最接近的设备编号为" << dev << endl;
	cudaSetDevice(dev);
	getchar();
	return 0;
}