EADST

Jetson Check GPU Memory Usage with C++

Jetson Check GPU Memory Usage with C++

#include < iostream>
#include < unistd.h>
#include "cuda.h"

int main()
{
    // show memory usage of GPU
    size_t free_byte ;
    size_t total_byte ;

    while (true )
    {
        cudaError_t cuda_status = cudaMemGetInfo( &free_byte, &total_byte ) ;

        if ( cudaSuccess != cuda_status ){
            std::cout << "Error: cudaMemGetInfo fails, " << cudaGetErrorString(cuda_status) << std::endl;
            exit(1);
        }

        double free_db = (double)free_byte ;
        double total_db = (double)total_byte ;
        double used_db = total_db - free_db ;

        std::cout << "GPU memory usage: used = " << used_db/1024.0/1024.0 << ", free = "
                  << free_db/1024.0/1024.0 << " MB, total = " << total_db/1024.0/1024.0 << " MB" << std::endl;
        sleep(1);
    }

    return 0;
}

Reference

Nvidia Forums

相关标签
About Me
XD
Goals determine what you are going to be.
Category
标签云
Zip Data News CLAP 云服务器 RGB Baidu 第一性原理 VGG-16 Jetson Domain Windows Tiktoken Tracking Card Template Rebuttal Google PIP Markdown 版权 v0.dev Food Miniforge Jupyter WAN 关于博主 CEIR v2ray diffusers SVR C++ FP16 论文速读 LaTeX git IndexTTS2 Paper Proxy Numpy Docker 强化学习 LLAMA WebCrawler 搞笑 Algorithm Distillation git-lfs SPIE XGBoost Agent Quantize 阿里云 VPN ResNet-50 Excel Search Web Disk ms-swift Hotel FP32 域名 OpenAI Plotly FastAPI Nginx GoogLeNet UI Python PDF Quantization Clash COCO CAM Input printf Qwen CC Review Transformers HuggingFace Ptyhon 图标 Pickle Bitcoin Vim TensorRT ChatGPT ONNX CSV CTC Cloudreve uWSGI OpenCV Interview Streamlit Video NameSilo YOLO LeetCode MD5 mmap 净利润 PyCharm LoRA API Bert Freesound 多线程 LLM Dataset FlashAttention NLTK SAM Hungarian 报税 Claude 论文 JSON FP8 Heatmap Pandas Linux Math Django HaggingFace EXCEL Pillow CV ModelScope Translation GPTQ Random Hilton 音频 Firewall Augmentation Password torchinfo XML Knowledge Color scipy 财报 递归学习法 Breakpoint Qwen2.5 logger FP64 Statistics Datetime OCR tar Anaconda DeepSeek TensorFlow GIT llama.cpp SQLite 算法题 公式 GGML Github Animate Bin 多进程 Vmess Pytorch PDB InvalidArgumentError NLP AI Magnet Safetensors Paddle Mixtral transformers Website QWEN Michelin Logo Ubuntu RAR Bipartite DeepStream Permission 飞书 图形思考法 BeautifulSoup tqdm Tensor Use Sklearn Image2Text Diagram CUDA uwsgi BTC Shortcut 证件照 Base64 Land Conda TSV GPT4 Attention PyTorch 签证 顶会 hf Qwen2 UNIX SQL Git 继承 腾讯云 icon Crawler BF16 Gemma Plate Llama VSCode TTS
站点统计

本站现有博文330篇,共被浏览860961

本站已经建立2569天!

热门文章
文章归档
回到顶部