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

本站现有博文333篇,共被浏览913698

本站已经建立2616天!

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