cuda编程入门示例2---CUDA environment initialization

最新推荐文章于 2024-03-31 03:33:36 发布

原创最新推荐文章于 2024-03-31 03:33:36 发布 · 853 阅读

0 ·

CC 4.0 BY-SA版权

c/c++ 同时被 2 个专栏收录

66 篇文章

订阅专栏

cuda编程常用代码示例

25 篇文章

订阅专栏

本文介绍了一个简单的CUDA初始化过程及错误处理方法。通过检测设备数量并选择支持CUDA 1.x版本的设备，确保CUDA能够正常运行。此外，还提供了一系列自定义宏用于更方便地进行错误检查。

部署运行你感兴趣的模型镜像

#include <stdio.h>
#include <stdlib.h>
#include <cuda_runtime.h>

#define BLOCK_SIZE 16
static void HandleError(cudaError_t err, const char *file, int line)
{
	if (err != cudaSuccess)
	{
		printf("%s in %s at line %d\n", cudaGetErrorString(err), file, line);
		exit(EXIT_FAILURE);
	}
}
#define HANDLE_ERROR( err ) (HandleError( err, __FILE__, __LINE__ ))

#define HANDLE_NULL( a ) {if ((a) == NULL) { \
	printf("Host memory failed in %s at line %d\n", \
	__FILE__, __LINE__); \
	exit(EXIT_FAILURE); }}

static bool InitCUDA()
{
	int count;

	cudaGetDeviceCount(&count);
	if (count == 0)
	{
		fprintf(stderr, "There is no device.\n");
		return false;
	}

	int i;
	for (i = 0; i < count; i++)
	{
		cudaDeviceProp prop;
		if (cudaGetDeviceProperties(&prop, i) == cudaSuccess)
		{
			if (prop.major >= 1)
			{
				break;
			}
		}
	}

	if (i >= count)
	{
		fprintf(stderr, "There is no device supporting CUDA 1.x.\n");
		return false;
	}

	cudaSetDevice(i);

	return true;
}

int main(int argc, char *argv[])
{
	if (!InitCUDA())
	{
		return -1;
	}
	printf("CUDA initialized.\n");

	//to do your own thing

	//remember to release the device
	cudaDeviceReset();

	return 0;
}

您可能感兴趣的与本文相关的镜像