以编程方式查询GPU设备属性

拼搏现实的明天。 2022-11-20 01:54 143阅读 0赞

由于 GPU 上的 SM 数量会因所用的特定 GPU 而异，因此为支持可移植性，您不得将 SM 数量硬编码到代码库中。相反，应该以编程方式获取此信息。

以下所示为在 CUDA C/C++ 中获取 C 结构的方法，该结构包含当前处于活动状态的 GPU 设备的多个属性，其中包括设备的 SM 数量：

int deviceId;
    cudaGetDevice(&deviceId);                  // `deviceId` now points to the id of the currently active GPU.
    
    cudaDeviceProp props;
    cudaGetDeviceProperties(&props, deviceId); // `props` now has many useful properties about
                                               // the active GPU device.

#include <stdio.h>
    
    int main()
    {
      /*
       * Device ID is required first to query the device.
       */
    
      int deviceId;
      cudaGetDevice(&deviceId);
    
      cudaDeviceProp props;
      cudaGetDeviceProperties(&props, deviceId);
    
      /*
       * `props` now contains several properties about the current device.
       */
    
      int computeCapabilityMajor = props.major;
      int computeCapabilityMinor = props.minor;
      int multiProcessorCount = props.multiProcessorCount;
      int warpSize = props.warpSize;
    
    
      printf("Device ID: %d\nNumber of SMs: %d\nCompute Capability Major: %d\nCompute Capability Minor: %d\nWarp Size: %d\n", deviceId, multiProcessorCount, computeCapabilityMajor, computeCapabilityMinor, warpSize);
    }

Tesla V100 的设备属性:

Device ID: 0  
Number of SMs: 80  
Compute Capability Major: 7  
Compute Capability Minor: 0  
Warp Size: 32

发表评论取消回复

表情：

评论列表（有 0 条评论，143人围观）

还没有评论，来说两句吧...

相关阅读

相关使用 React 路由器以编程方式导航

问：使用 react-router，我可以使用 Link 元素创建由 react 路由器本地处理的链接。我在内部看到它调用 this.context.transit

╰+攻爆jí腚メ/ 2024年03月29日 15:14/ 0 赞/ 60 阅读

相关关于ubuntu tensorflow-gpu2.x无法使用gpu设备

最近更新系统，也顺便升级了一下显卡驱动，由原来的418升级到455，随后使用tensorflow-gpu2.1无法使用gpu加速计算，一跑网络，cpu占用率接近100%。但是g

不念不忘少年蓝@/ 2022年12月26日 11:24/ 0 赞/ 181 阅读

相关查询计算机上是否有可用的GPU设备

查询计算机上是否有可用的GPU设备代码结果展示代码 def is_gpu_available(cuda_only=True):

Myth丶恋晨/ 2022年12月24日 15:57/ 0 赞/ 106 阅读

相关 Tensorflow跨设备调用GPU

注意：在开始讲述之前首先对`Tensorflow`框架下的模型的跨设备操作进行说明，具体跨设备操作包括以下四种情况： 1. `CPU`模型训练 -> `GPU`模型推理 2

谁践踏了优雅/ 2022年12月08日 12:50/ 0 赞/ 162 阅读

相关我们如何以编程方式检测运行设备的iOS版本？ [重复]

本文翻译自：[How can we programmatically detect which iOS version is device running on? \[dupl

我不是女神ヾ/ 2022年11月27日 07:12/ 0 赞/ 114 阅读

相关 UIScrollView以编程方式滚动到底部

本文翻译自：[UIScrollView scroll to bottom programmatically][] How can I make a `UIScrollView

小咪咪/ 2022年11月25日 13:10/ 0 赞/ 213 阅读

相关 JS中以变量方式访问对象属性

this.enterprise\[val\] 如 search(){ this.$api.enterpriseDetail(this.params).t

末蓝、/ 2022年11月22日 00:11/ 0 赞/ 191 阅读

相关以编程方式查询GPU设备属性

由于 GPU 上的 SM 数量会因所用的特定 GPU 而异，因此为支持可移植性，您不得将 SM 数量硬编码到代码库中。相反，应该以编程方式获取此信息。以下所示为在 CUDA

拼搏现实的明天。/ 2022年11月20日 01:54/ 0 赞/ 144 阅读

相关【blkid】查询或查看磁盘设备的相关属性

【blkid】查询或查看磁盘设备的相关属性 \[root@gdy ftpftp\]\ blkid /dev/sda2 /dev/sda2: UU

Love The Way You Lie/ 2022年07月27日 10:51/ 0 赞/ 98 阅读

相关 pytorch指定使用的gpu设备

最近在使用pytorch写程序，想在指定的GPU设备上执行。假设函数func( )使用到了GPU设备：则在func()函数内部或者调用func()的地方加上:

末蓝、/ 2022年04月02日 15:50/ 0 赞/ 284 阅读