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;
相關文章
相關標籤/搜索