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

1 visualización (últimos 30 días)
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

Respuestas (0)

Categorías

Más información sobre Image Data Workflows en Help Center y File Exchange.

Community Treasure Hunt

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

Start Hunting!

Translated by