利用CUDA查看多张显卡可用显存和总显存大小

利用CUDA查看每张显卡上的可用显存大小和总的显存大小，参考了博文1，博文2，主要使用的函数是cudaMemGetInfo()，cudaGetDeviceCount()和cudaSetDevice()

#include <cuda.h>
#include <stdio.h>
#include <cuda_runtime.h>
#include <device_launch_parameters.h>int main()
{size_t avail;size_t total;int deviceCount=0;cudaGetDeviceCount(&deviceCount);              // 用deviceCount获取显卡总数量for(int i_dev=0;i_dev<deviceCount;i_dev++){cudaSetDevice(i_dev);                      // 使用第i_dev张显卡作为使用的显卡cudaMemGetInfo(&avail, &total);            // 获取可用和总显存大小printf("Device %d Memeory:\n",i_dev);printf("Avaliable Memery = %dm   Total Memory = %dm\n", int(avail/1024/1024), int(total / 1024 / 1024));printf("\n");      }
}

博文2中使用的是 cuInit(),cuMemGetInfo()函数，在linux下编译的时候需要加上 -lcuda 不然会报错，这篇博文的代码好像有点问题，就是第一个cuMemGetInfo()获取的数据都为0。

size_t 获取的显卡内存单位是字节(B)，数据大小以及超过int的范围了，要么用long long型接收显存大小的数据，要么先处理转换成兆字节（MB）

利用CUDA查看多张显卡可用显存和总显存大小相关推荐

Ubuntu 16.04系统中利用CUDA安装更新NVIDIA显卡驱动程序的方法
严正声明:本文系作者davidhopper原创,未经许可,不得转载. 在ThinkPad系列笔记本电脑中使用过Ubuntu 16.04系统的同学都知道,NVIDIA显卡驱动程序特别难安装.以我使用的T ...
Linux/ubuntu深度学习，查看cuda、cudnn、显卡、驱动的版本的命令
查看Linux/ubuntu深度学习的cuda.cudnn.显卡.驱动的版本 1.查看cudnn版本: 法一: nvcc -V 注意:V是大写的. 法二: cat /usr/local/cuda/ve ...
CUDA：利用Pytorch查看自己电脑上CUDA版本及其相关信息
CUDA:利用Pytorch查看自己电脑上CUDA版本及其相关信息目录利用Pytorch查看自己电脑上CUDA的版本信息
计算机的主板显卡内存条怎么查,电脑显卡在哪看?查看自己电脑显卡的显存等信息的方法...
显卡的显存相当于电脑的内存一样,用来存储显示数据信息,很多朋友新买来的电脑,我们有时候想要查看自己电脑显卡的显存大小却不知怎么看,那么,电脑显卡在哪看呢?下面小编分享下查看自己电脑显卡显存信息的方法, ...
pytorch指定用多张显卡训练_Pytorch中多GPU训练指北
前言在数据越来越多的时代,随着模型规模参数的增多,以及数据量的不断提升,使用多GPU去训练是不可避免的事情.Pytorch在0.4.0及以后的版本中已经提供了多GPU训练的方式,本文简单讲解下使用P ...
Keras同时用多张显卡训练网络
正文共4542个字,预计阅读时间12分钟. Author: Zongwei Zhou | 周纵苇 Weibo: @MrGiovanni Email: zongweiz@asu.edu Referenc ...
mysql查询一个表有哪些索引_如何查看某张数据库表上都有哪些索引（转）
索引使用简介一. 关于索引的知识要写出运行效率高的sql,需要对索引的机制有一定了解,下面对索引的基本知识做一介绍. 1. 索引的优点和局限索引可以提高查询的效率,但会降低dml操作的效率. 所 ...
利用CUDA流重叠计算和数据传输
利用CUDA流重叠计算和数据传输 CUDA中有一个重要的概念是流(stream). 其实它代表着一系列的指令的执行队列. 这个执行队列就像他的名字一样, 有着固定的执行顺序(就像河流只能向一个方向固 ...
利用redis-sentinel+keepalived实现redis高可用
目标.需求: 为上层应用提供高可靠.低延迟.低(无限接近0)数据损失的Redis缓存服务方案概述: 采用同一网络内的三台主机(可以是物理主机.虚拟机或docker容器),要求三台主机之间都能相互访问 ...

利用CUDA查看多张显卡可用显存和总显存大小

利用CUDA查看多张显卡可用显存和总显存大小相关推荐

最新文章

热门文章