CUDA 设备操作实例

张大饼的最爱

已于 2024-10-14 16:41:11 修改

阅读量328

点赞数 3

CC 4.0 BY-SA版权

分类专栏： CUDA 文章标签： cuda c CUDA

于 2024-10-14 16:40:14 首次发布

本文链接：https://blog.csdn.net/Jane_yuhui/article/details/142922602

CUDA 专栏收录该内容

15 篇文章

订阅专栏

我的电脑上只有一个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;
}

结果：