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

本站现有博文327篇,共被浏览833334

本站已经建立2538天!

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