1. 利用gpuArray()函数将数据从CPU传入GPU中
a=zeros(2,3);
a=gpuArray(a);
或
a=gpuArray(single(a));%数据默认都是双精度,降成单精度会提高计算速度
2.用class()可以查看数据类型
>> a=zeros(2,3);
>> class(a)
ans =
'double'
>> a=gpuArray(a);
>> class(a)
ans =
'gpuArray'
3. 数据在进行运算时,只要有一个变量在GPU上,其他变量也会自动进入 GPU一起运算,产生的结果也在GPU上。
>> a=gpuArray([1 2 3]);
>> b=[1 2 3];
>> c=a+b;
>> class(a)
ans =
'gpuArray'
>> class(b)
ans =
'double'
>> class(c)
ans =
'gpuArray'
4.如果需要的话,用gather()函数将数据从GPU传回CPU
>> a=gpuArray([1 2 3]);
>> class(a)
ans =
'gpuArray'
>> a=gather(a);
>> class(a)
ans =
'double'
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)