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

本站现有博文322篇,共被浏览791850

本站已经建立2488天!

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