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

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

本站已经建立2534天!

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