DNN classify function uses GPU, or just CPU ?
2 vues (au cours des 30 derniers jours)
Afficher commentaires plus anciens
I have been using Deep Learning for a short time; I learned to use the GPU for network training.
When I trained the neural network, I use the classify function to make it "work" on the images that the network has to classify.
Does anyone know whether, in this working phase, "classify" uses GPU or CPU ?
I couldn't find any information in the documentation, and the CPU/GPU setting that sets the training seems to me to have no effect in the working phase.
Thanks in advance
0 commentaires
Réponse acceptée
Chunru
le 6 Juil 2022
Classify will automatically choose GPU (if it is available and suitable).
You can also specify the use of GPU explicitly:
classify(net, X, 'ExecutionEnvironment', 'gpu');
% Available options: 'auto', 'gpu', 'cpu', 'multi-gpu', 'parallel'
% default is 'auto'
Plus de réponses (0)
Voir également
Catégories
En savoir plus sur Deep Learning Toolbox dans Help Center et File Exchange
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!