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

조회 수: 17 (최근 30일)
Prb
Prb 2019년 5월 14일
댓글: Prb 2019년 5월 15일
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일
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개)

카테고리

Help CenterFile Exchange에서 GPU Computing에 대해 자세히 알아보기

Community Treasure Hunt

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

Start Hunting!

Translated by