matlab使用GPU加速方法
查看服务器GPU计算资源可以用:
nvidia-smi
查看空闲的GPU是哪块,之后可以指定该GPU。
就算都空闲,也最好指定下,实验室资源一般有限,别全占了,会被说(资源无限的随意,不用指定,随便用)
用下面命令指定GPU.
g=gpuDevice(1)
此时指定的是第0个GPU,matlab是从1,2,3,4开始计数的,GPU是从0,1,2,3计数的。
使用GPU用指令gpuArray
A1 = gpuArray(rand(2000,2000));
B1 = gpuArray(rand(2000,2000));
C1 = gpuArray(rand(2000,2000));
D1=A1+B1;E1=A1.*D1;
此时会调用GPU资源进行计算。
计算后再用gather指令将GPU数据转为CPU数据
cc=gather(D1)