Я пишу программу для тренировки моих данных (изображения с trainNetwork).Для обучения я использую компьютер с графическим процессором nvidia quadpro M4000 и оперативной памятью 64 ГБ.Я пишу короткий код здесь:
opts=trainingOptions('sgdm','InitialLearnRate',1e-3,'MaxEpochs',200,'MiniBatchSize',64,'Plots','training-progress', 'ExecutionEnvironment', 'auto');
net=trainNetwork(trainingData,layers,opts);
Как описано на сайте Mathworks, trainNetwork
выберет графический процессор, когда он будет доступен.Хотя я пытался изменить ExecutionEnvironment
на GPU, он все еще работает только на процессоре.Как заставить trainNetwork
работать на GPU?