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

本站现有博文339篇,共被浏览996025次

本站已经建立2706天!

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