CUDA 设备操作实例

我的电脑上只有一个GPU,下面的例子就是先获取GPU的设备数量,然后设置GPU执行时使用的设备。

#include "cuda_runtime.h"

#include  <stdio.h>

int main()
{
	int nDeviceCount = 0;
	cudaError_t error = cudaGetDeviceCount(&nDeviceCount);

	if (error != cudaSuccess || nDeviceCount == 0)
	{
		return -1;
	}
	printf("The count of GPUs is %d.\n", nDeviceCount);

	// 设置执行 
	int iDevc = 0;
	error = cudaSetDevice(iDevc);
	if (error != cudaSuccess)
	{
		return -1;
	}

	printf("Set GPU 0 for computing.\n");

	getchar();

	return 0;
}

结果:

上一篇:Python 无法安装pybluez 解决办法


下一篇:运用 JDK8 中的核心新特性