CUDA - GPU Memory Usage Check

// show memory usage of GPU

size_t free_byte ;

size_t total_byte ;

cuda_status = cudaMemGetInfo( &free_byte, &total_byte ) ;

if ( cudaSuccess != cuda_status )

{

printf("Error: cudaMemGetInfo fails, %s \n", cudaGetErrorString(cuda_status) );

exit(1);

}

double free_db = (double)free_byte ;

double total_db = (double)total_byte ;

double used_db = total_db - free_db ;

printf("GPU memory usage: used = %f, free = %f MB, total = %f MB\n",

used_db/1024.0/1024.0, free_db/1024.0/1024.0, total_db/1024.0/1024.0);

출처 : https://devtalk.nvidia.com/default/topic/487541/best-way-to-report-memory-consumption-in-cuda-/?offset=1