清空筛选条件
清空筛选条件

How can we train using gpu instead of cpu ?

14 次查看（过去 30 天）
Hamza Afzal
Hamza Afzal 2021-2-26
评论： Joss Knight 2021-3-8
Data:
https://www.mathworks.com/help/vision/ug/object-detection-using-yolo-v3-deep-learning.html
I am using this example, i want to train the network using gpu.
Problem:
Where in the code, i can specify that training should be done using gpu?
Is there a code line or any functions that does this ?

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

采纳的回答

Joss Knight
Joss Knight 2021-3-8
By default your GPU will be used if you have one. To force it, set the ExecutionEnvironment training option to 'gpu'.
  2 个评论
Hamza Afzal
Hamza Afzal 2021-3-8
ExecutionEnvironment = 'gpu'
Do we have to write this code line in the code(program) ?
Joss Knight
Joss Knight 2021-3-8
Ah, my mistake. This is a custom training loop. In this case, the simplest way to force GPU behaviour is to set the OutputEnvironment option on the minibatchqueue object mbqTrain when it is created. For best performance, you should also move the dlnetwork object to the GPU:
net = dlupdate(@gpuArray, net);
You can do this inside the training code, just before if doTraining.
But what I said before holds true - this training code will run on the GPU as it is because that is the way the default settings work.
For inference, you can see that the section Detect objects Using YOLO v3 takes pains to show you how to run on the GPU.

请先登录，再进行评论。

更多回答（0 个）

类别

在 Help Center 和 File Exchange 中查找有关 Deep Learning Toolbox 的更多信息

标签

产品


版本

R2020a

Community Treasure Hunt

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

Start Hunting!

Translated by