代码之家  ›  专栏  ›  技术社区  ›  Freelancer

如何在GPU上强制MATLAB运行CNN

  •  1
  • Freelancer  · 技术社区  · 5 年前

    我写了一个程序来训练我的数据(图像与trainNetwork)。我用带GPU的电脑 内存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 列车网络

    0 回复  |  直到 5 年前