欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

matlab下GPU编程笔记

程序员文章站 2022-06-09 12:34:41
...
不多说,直接上代码
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





另外也可以看看:http://www.360doc.com/content/14/0311/10/13256259_359504252.shtml
相关标签: matlab