我编写了一个程序来训练我的数据(使用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上运行?