CUDA编程:获取GPU总显存和可用显存

代码很简单: 就是使用函数:cudaMemGetInfo( &avail, &total );ios #include <cuda_runtime.h> #include <helper_cuda.h> #include <iostream> #include <memory> #include <string> void checkGpuMem() { size_t avail;
相关文章
相关标签/搜索