can we assign number of CPUs or number of GPUs should be working in DNN ?

Question

Furat Alobaidy 2021-2-8

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/738962-can-we-assign-number-of-cpus-or-number-of-gpus-should-be-working-in-dnn

回答： Walter Roberson 2021-2-8

Hi ,

please i need running image classification for DNN by CPUs and GPUs , so is there specif command in Matlab can i assign how many CPUs or GPUs should be worked for doing the classification DNN application ?.

0 个评论
显示 -2更早的评论隐藏 -2更早的评论

请先登录，再进行评论。

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

Answer 1

Walter Roberson 2021-2-8

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/738962-can-we-assign-number-of-cpus-or-number-of-gpus-should-be-working-in-dnn#answer_617292

My understanding is that only one GPU will be used. The interface to GPUs only permits one GPU to be assigned per process -- if a process switches between GPUs then the current GPU calculation for the process gets reset and discarded.

To control the maximum number of CPUs, see https://www.mathworks.com/help/matlab/ref/maxnumcompthreads.html or create a parpool that is configured for the appropriate number of CPUs.