头文件
#include <cuda.h>
#include <cuda_runtime.h>
#include <device_launch_parameters.h>
调用函数
cuda::setDevice(0);
size_t free;
size_t total;
CUresult uRet = cuMemGetInfo(&free, &total);
if (uRet == CUDA_SUCCESS)
printf("free = %dM\ntotal = %dM", free / 1024 / 1024, total / 1024 / 1024);
结果: