DNN classify function uses GPU, or just CPU ?
조회 수: 2 (최근 30일)
이전 댓글 표시
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
채택된 답변
Chunru
2022년 7월 6일
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'
추가 답변 (0개)
참고 항목
카테고리
Help Center 및 File Exchange에서 Deep Learning Toolbox에 대해 자세히 알아보기
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!