Как заставить MATLAB запустить CNN на GPU - PullRequest
1 голос
/ 01 апреля 2019

Я пишу программу для тренировки моих данных (изображения с 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?

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...