matlab下GPU编程笔记

最新推荐文章于 2025-06-28 07:03:09 发布

原创最新推荐文章于 2025-06-28 07:03:09 发布 · 615 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#matlab

Matlab 专栏收录该内容

16 篇文章

订阅专栏

本文介绍了如何在MATLAB中使用GPU进行高效计算，包括如何查看GPU设备、配置当前GPU、将数据移动到GPU和从GPU中获取数据，以及GPU上的基本运算。通过实例展示了如何在GPU中执行矩阵乘法等操作。

不多说，直接上代码

gpuDevice   %   查看系统中的gpu,,其中的DeviceSupported会给出matlab支持的GPU个数。
g=gpuDevice(1);    %会清空 GPU 1中的所有数据,,将GPU1 设为当前GPU
reset(g)  %也可以清空GPU中数据。

a=1;
a=gpuArray(a);    %将a从CPU移到GPU中
onGPU=existsOnGPU(a)
c=class(a)
uc=classUnderlying(a)  %a的类型，，在GPU中也可以使用，，此时class是失效的
a=gather(a);   %将a从GPU中移动到CPU中
% existsOnGPU(a)  %报错！！！

a=gpuArray(a);    %将a从CPU移到GPU中
reset(g);
onGPUAfterReset=existsOnGPU(a)  %reset后，虽然workspace中还有a，但是已经无效了。。。


%% 运算
a=gpuArray.ones(2,2,'single');
b=ones(1,1);
c=a*b;
class(c)  %gpuArray