Can I use cpu instead of gpu for the DeepLearni​ngImageCla​ssificatio​nExample.m

2017 5 30
1 个回答
更新时间：2020 4 17
8 次查看（30 天）

0 个投票

I was running the "DeepLearningImageClassificationExample.m" example but this failed because of unsupported GPU. I did see that this example requires a "CUDA-capable GPU card" but is there a manner to to use the parameters ['ExecutionEnvironment','cpu'] in the "activation" function, and use the CPU instead?

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

回答（1 个）

Joss Knight
Joss Knight 2017-5-31

0 个投票

Yes

2 个评论
显示 无 隐藏 无

Walter Roberson
Walter Roberson 2020-4-17
On line 68 change
trainingFeatures = activations(net, augmentedTrainingSet, featureLayer, ...
'MiniBatchSize', 32, 'OutputAs', 'columns');
to
trainingFeatures = activations(net, augmentedTrainingSet, featureLayer, ...
'MiniBatchSize', 32, 'OutputAs', 'columns', 'ExecutionEnvironment', 'cpu');

请先登录，再进行评论。

类别

在 帮助中心 和 File Exchange 中查找有关 Deep Learning Toolbox 的更多信息

标签

Community Treasure Hunt

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

Start Hunting!

Translated by