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

2 vues (au cours des 30 derniers jours)
Tommaso Di Noto
Tommaso Di Noto le 29 Août 2017
Modifié(e) : Edric Ellis le 30 Août 2017
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
  1 commentaire
Joss Knight
Joss Knight le 29 Août 2017
What is your version of MATLAB and what GPU do you have? Have you installed the latest NVIDIA driver for that device?

Connectez-vous pour commenter.

Connectez-vous pour répondre à cette question.

Réponses (0)

Catégories

En savoir plus sur Image Data Workflows dans Help Center et File Exchange

Tags

Community Treasure Hunt

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

Start Hunting!

Translated by