Error using parallel.gpu.CUDADevice.hBuild during the training of a Convolutional Neural Network
4 Ansichten (letzte 30 Tage)
Ältere Kommentare anzeigen
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 Kommentar
Joss Knight
am 29 Aug. 2017
What is your version of MATLAB and what GPU do you have? Have you installed the latest NVIDIA driver for that device?
Antworten (0)
Siehe auch
Kategorien
Mehr zu GPU Computing finden Sie in Help Center und File Exchange
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!