matlab下GPU编程笔记
程序员文章站
2022-06-09 12:34:41
...
不多说,直接上代码
另外也可以看看:http://www.360doc.com/content/14/0311/10/13256259_359504252.shtml
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