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

本站现有博文337篇,共被浏览952631次

本站已经建立2663天!

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