win10 + vs2015 + cuda8.0 读取显存使用情况

最新推荐文章于 2024-05-22 22:47:43 发布

JoannaJuanCV

最新推荐文章于 2024-05-22 22:47:43 发布

阅读量1.6k

点赞数 2

分类专栏：图像处理算法（c++/python opencv）文章标签：显存使用率

本文链接：https://blog.csdn.net/zfjBIT/article/details/84973708

版权

图像处理算法（c++/python opencv）专栏收录该内容

90 篇文章 6 订阅

订阅专栏

首先要在vs2015上配置好cuda环境；

头文件：

#include <cuda.h>
#include <cuda_runtime.h>
#include <device_launch_parameters.h>

代码段：

CUresult uRet;
//int *dn;
//cudaMalloc((void **)&dn, 1024 * 1024 * 100 * sizeof(int));
uRet = cuInit(0);
//printf("%d\n", (int)uRet);
size_t free;
size_t total;
uRet = cuMemGetInfo(&free, &total);
if (uRet == CUDA_SUCCESS)
	printf("\nfree = %d\ntotal = %d\n", free / 1024 / 1024, total / 1024 / 1024);
		
printf("\ncudaMalloc....\n");
int *dn;
cudaMalloc((void **)&dn, 1024 * 1024 * 100 * sizeof(int));
		
size_t free1;
size_t total1;
uRet = cuMemGetInfo(&free1, &total1);
if (uRet == CUDA_SUCCESS)
	printf("\nfree = %d\ntotal = %d\n", free1 / 1024 / 1024, total1 / 1024 / 1024);

printf("\ncudaMalloc Size = %dM\n", (free - free1)/1024/1024);

本机测试：

显卡类型：NVIDIA GeForce GTX 1060 6GB

结果：