matlab 调用GPU运算

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'