Run GPU code only on CPU: How to get rid of gpuArray data?

Prb
2019 5 14
1 个回答
回答已采纳
更新时间：2019 5 15
1 次查看（30 天）

0 个投票

Hi I have this piece of deep learning code which runs on GPU. I wish not to use GPU. How can I change the code to run only on CPU or just to use it as Array.
eg:
B_forward = gpuArray(X_test(:,:,:,ThisInds));
if size(Residuals,1) > size(B_forward,1)
NewRes = int8(zeros(size(B_forward),'gpuArray'));
end
if Shifts(WeightLayerIndex) ~= 0
B_forward = gpuArray(bitshift(gather(B_forward),Shifts(WeightLayerIndex)));
end

请先登录，再回答此问题。

 采纳的回答

Walter Roberson
Walter Roberson 2019-5-14

1 个投票

B_forward = (X_test(:,:,:,ThisInds));
if size(Residuals,1) > size(B_forward,1)
NewRes = int8(zeros(size(B_forward)));
end
if Shifts(WeightLayerIndex) ~= 0
B_forward = (bitshift((B_forward),Shifts(WeightLayerIndex)));
end

更多回答（0 个）

类别

在 帮助中心 和 File Exchange 中查找有关 GPU Computing 的更多信息

标签

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by