只要你能使用cuda,那么就可以直接使用頭文件：?#include "cuda_runtime_api.h"

以下代碼通過(guò)該api返回了當(dāng)前顯卡的總共的顯存容量、已使用顯存容量、剩余顯存容量

#include <iostream> #include "cuda_runtime_api.h"static void GetGPUMemory() {int deviceCount = 0;cudaError_t error_id = cudaGetDeviceCount(&deviceCount);if (deviceCount == 0){std::cout << "當(dāng)前PC沒(méi)有支持CUDA的顯卡硬件設(shè)備" << std::endl;}size_t gpu_total_size;size_t gpu_free_size;cudaError_t cuda_status = cudaMemGetInfo(&gpu_free_size, &gpu_total_size);if (cudaSuccess != cuda_status) {std::cout << "Error: cudaMemGetInfo fails : " << cudaGetErrorString(cuda_status) << std::endl;exit(1);}double total_memory = double(gpu_total_size) / (1024.0 * 1024.0);double free_memory = double(gpu_free_size) / (1024.0 * 1024.0);double used_memory = total_memory - free_memory;std::cout << "\n"<< "當(dāng)前顯卡總共有顯存" << total_memory << "m \n"<< "已使用顯存" << used_memory << "m \n"<< "剩余顯存" << free_memory << "m \n" << std::endl;}int main() {GetGPUMemory();return 0; }

輸出：

?

ref :?C++ - C++使用cuda api獲取當(dāng)前GPU顯卡的總共的顯存容量、已使用顯存容量、剩余顯存容量 - StubbornHuang Blog

總結(jié)

以上是生活随笔為你收集整理的如何查看当前位置显存使用情况的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。