Error using parallel.g​pu.CUDADev​ice.hBuild during the training of a Convolutional Neural Network

1 次查看(过去 30 天)
Hello! I have this code:
layers = [imageInputLayer([28 28 1])
convolution2dLayer(5,20)
reluLayer
maxPooling2dLayer(2,'Stride',2)
fullyConnectedLayer(10)
softmaxLayer
classificationLayer()]; %define the Convolutional Neural Network (CNN)
options = trainingOptions('sgdm','MaxEpochs',15, 'InitialLearnRate',0.0001); %set the training options
convnet = trainNetwork(trainDigitData,layers,options); %train my CNN
and when I train the network I get all these errors:
  • Error using parallel.gpu.CUDADevice.hBuild
  • Error in parallel.gpu.GPUDevice.getDevice
  • Error in parallel.gpu.GPUDevice.isAvailable
  • Error in nnet.internal.cnn.util.isGPUCompatible
  • Error in nnet.internal.cnn.util.GPUShouldBeUsed
  • Error in trainNetwork>iSetupExecutionEnvironment
  • Error in trainNetwork

回答(0 个)

类别

Help CenterFile Exchange 中查找有关 Image Data Workflows 的更多信息

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by