Obtain memory usage and class for GPU variable

Question

Peter 2016-5-2

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/282129-obtain-memory-usage-and-class-for-gpu-variable

评论： Peter 2021-8-4

How can I get the amount of memory space (bytes) and the class (single/double/int) of a GPU variable? 'whos' doens't provide what I need. E.g.:

A=rand(1000,gpuArray);
infoA=whos('A');

gives: infoA =

          name: 'A'
          size: [1000 1000]
         bytes: 108
         class: 'gpuArray'
        global: 0
        sparse: 0
       complex: 0
       nesting: [1x1 struct]
    persistent: 0

but i still don't know how many bytes of GPU memory A is taking and whether A is a single / double etc.

0 个评论
显示 -2更早的评论隐藏 -2更早的评论

请先登录，再进行评论。

请先登录，再回答此问题。

Answer 1

Matt J 2021-3-1

1
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/282129-obtain-memory-usage-and-class-for-gpu-variable#answer_636194

See whosGPU.

1 个评论
显示 -1更早的评论隐藏 -1更早的评论

Peter 2021-8-4

awesome, thanks!

请先登录，再进行评论。

Answer 2

Edric Ellis 2016-5-3

2
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/282129-obtain-memory-usage-and-class-for-gpu-variable#answer_220573

编辑：Edric Ellis 2016-5-3

在 MATLAB Online 中打开

Unfortunately, whos only reports the memory usage on the CPU of a gpuArray. For non-sparse gpuArray data, you can compute the number of bytes consumed like so:

dataType = classUnderlying(A);
switch dataType
  case 'double'
    bytesPerElem = 8;
  case 'single'
    bytesPerElem = 4;
  case 'logical'
    bytesPerElem = 1;
  otherwise
    bytesPerElem = str2double(regexprep(dataType, '[a-z]', '')) / 8;
end
if ~isreal(A)
  bytesPerElem = 2 * bytesPerElem;
end
totalBytes = bytesPerElem * numel(A);