如何强制MATLAB在GPU上运行CNN

时间:2019-04-01 08:18:09

标签: matlab deep-learning

我编写了一个程序来训练我的数据(使用trainNetwork拍摄的图像)。我使用一台具有GPU nvidia quadpro M4000 和RAM 64GB的计算机进行训练。我在这里写一个简短的代码:

opts=trainingOptions('sgdm','InitialLearnRate',1e-3,'MaxEpochs',200,'MiniBatchSize',64,'Plots','training-progress', 'ExecutionEnvironment', 'auto');
net=trainNetwork(trainingData,layers,opts);

作为Mathworks网站上的描述,trainNetwork将在可用时选择GPU。尽管我尝试将ExecutionEnvironment更改为gpu,但它仍然只能在CPU上运行。如何强制trainNetwork在GPU上运行?

0 个答案:

没有答案