惯性聚合 高效追踪和阅读你感兴趣的博客、新闻、科技资讯
阅读原文 在惯性聚合中打开

推荐订阅源

IT之家
IT之家
OSCHINA 社区最新新闻
OSCHINA 社区最新新闻
A
About on SuperTechFans
博客园 - 聂微东
奇客Solidot–传递最新科技情报
奇客Solidot–传递最新科技情报
freeCodeCamp Programming Tutorials: Python, JavaScript, Git & More
B
Blog RSS Feed
U
Unit 42
Stack Overflow Blog
Stack Overflow Blog
Recent Announcements
Recent Announcements
雷峰网
雷峰网
罗磊的独立博客
Microsoft Security Blog
Microsoft Security Blog
Hugging Face - Blog
Hugging Face - Blog
L
LangChain Blog
人人都是产品经理
人人都是产品经理
The GitHub Blog
The GitHub Blog
F
Fortinet All Blogs
Cyber Security Advisories - MS-ISAC
Cyber Security Advisories - MS-ISAC
H
Help Net Security
P
Proofpoint News Feed
The Cloudflare Blog
D
Docker
大猫的无限游戏
大猫的无限游戏

博客园 - yeren2046

把 CodeBuddy 接进 DSH RedCoins,一个免费的类似bluecoins的个人财务管理软件 记录一个栈溢出导致的崩溃问题 一个免费的图片数据标注工具 常用zip命令 git版本导致的"Permission denied (publickey). fatal: Could not read from remote repository." 只显示全部特定进程名的top信息的shell脚本 linux 上用 core 文件定位线上问题 ffmpeg 时基转换 昇腾卡通道号范围 ffmpeg视频截取 英伟达硬解码错误汇总 DVPP问题汇总 TensorRT生成INT8校准文件 结构体指定初始化 ffmpeg命令行基于英伟达显卡编解码的转码 C++11 获取当前时间戳 nvjpeg 简单使用 AV_PIX_FMT_CUDA 数据转 RGB C++ do{ } while(0)
基于CUDA查询显卡型号和显存大小
yeren2046 · 2022-09-27 · via 博客园 - yeren2046

安装驱动后,我们可以用 nvidia-smi 命令来获取显卡信息。

nvidia-smi 的一个比较好的博客 nvidia-smi详解

nvidia-smi

Tue Sep 27 11:08:37 2022       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 515.65.01    Driver Version: 515.65.01    CUDA Version: 11.7     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|                               |                      |               MIG M. |
|===============================+======================+======================|
|   0  NVIDIA GeForce ...  Off  | 00000000:3B:00.0 Off |                  N/A |
| 38%   53C    P2   104W / 320W |   3839MiB / 10240MiB |      0%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+
|   1  NVIDIA GeForce ...  Off  | 00000000:86:00.0 Off |                  N/A |
| 30%   36C    P8    19W / 320W |     13MiB / 10240MiB |      0%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+
|   2  NVIDIA GeForce ...  Off  | 00000000:AF:00.0 Off |                  N/A |
| 30%   34C    P8    13W / 320W |     13MiB / 10240MiB |      0%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+
| Processes:                                                                  |
|  GPU   GI   CI        PID   Type   Process name                  GPU Memory |
|        ID   ID                                                   Usage      |
|=============================================================================|
|    0   N/A  N/A      1398      G   /usr/lib/xorg/Xorg                  4MiB |
|    0   N/A  N/A      2340      C   java                             3826MiB |
|    0   N/A  N/A      2680      G   /usr/lib/xorg/Xorg                  4MiB |
|    1   N/A  N/A      1398      G   /usr/lib/xorg/Xorg                  4MiB |
|    1   N/A  N/A      2680      G   /usr/lib/xorg/Xorg                  4MiB |
|    2   N/A  N/A      1398      G   /usr/lib/xorg/Xorg                  4MiB |
|    2   N/A  N/A      2680      G   /usr/lib/xorg/Xorg                  4MiB |
+-----------------------------------------------------------------------------+

列出可用设备

nvidia-smi -L

GPU 0: NVIDIA GeForce RTX 3080 (UUID: GPU-d6daa455-a57f-f91a-a780-19d2a2c3e059)
GPU 1: NVIDIA GeForce RTX 3080 (UUID: GPU-aff98a0d-53bd-a538-d35f-70e0714cb8d2)
GPU 2: NVIDIA GeForce RTX 3080 (UUID: GPU-b1f86dba-045e-01a8-6e08-d170c25663da)

基于cuda获取显卡信息:

#define checkCudaErrors(S) do {CUresult  status; \
        status = S; \
        if (status != CUDA_SUCCESS ) std::cout << __LINE__ <<" checkCudaErrors - status = " << status << std::endl; \
        } while (false)

int CheckCUDAProperty( int devId )
{
    cuInit(0);

    CUdevice dev = devId;
    size_t memSize = 0;
    char devName[256] = {0};
    int major = 0, minor = 0;
    CUresult rlt = CUDA_SUCCESS;

    rlt = cuDeviceComputeCapability( &major, &minor, dev );
    checkCudaErrors( rlt );

    rlt = cuDeviceGetName( devName, sizeof( devName ), dev );
    checkCudaErrors( rlt );

    printf( "Using GPU Device %d: %s has SM %d.%d compute capability\n",
            dev, devName, major, minor );

    rlt = cuDeviceTotalMem( &memSize, dev );
    checkCudaErrors( rlt );

    printf( "Total amount of global memory:   %4.4f MB\n",
           (float)memSize / ( 1024 * 1024 ) );

    return 0;
}

运行结果:

Using GPU Device 1: NVIDIA GeForce RTX 3080 has SM 8.6 compute capability
Total amount of global memory:   10018.0625 MB