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

3 Ansichten (letzte 30 Tage)
Prb
Prb am 14 Mai 2019
Kommentiert: Prb am 15 Mai 2019
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

Akzeptierte Antwort

Walter Roberson
Walter Roberson am 14 Mai 2019
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

Weitere Antworten (0)

Kategorien

Mehr zu GPU Computing finden Sie in Help Center und File Exchange

Community Treasure Hunt

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

Start Hunting!

Translated by