osbm/cuda-kernel-nix
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

gpu memory script

Browse source
This commit is contained in:
Osman Faruk Bayram 2024-12-10 19:59:44 +03:00
parent c04da58ea3
commit 728d8c34c0
1 changed files with 12 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

25
main.cu
View file

@ -1,16 +1,13 @@
#include <iostream> #include <stdio.h>
#include "cuda.h"
extern "C"
int main() void checkGpuMem() {
{ float free_m, total_m, used_m;
int run_version, driver_version; size_t free_t,total_t;
std::cout<< "Return Code Runtime Version: "; cudaMemGetInfo(&free_t,&total_t);
std::cout<<cudaRuntimeGetVersion(&run_version); free_m =(uint)free_t/1048576.0 ;
std::cout<<"\nReturn Code Driver Version: "; total_m=(uint)total_t/1048576.0;
std::cout<<cudaDriverGetVersion(&driver_version); used_m=total_m-free_m;
std::cout << "\nRuntime Version: "; printf("GPU memory usage: used = %f, free = %f MB, total = %f MB\n",used_m,free_m,total_m);
std::cout << run_version;
std::cout << "\n Driver Version: ";
std::cout << driver_version;
return 0;
} }