Error using parallel.gpu.CUDADevice.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
1 个评论
Joss Knight
2017-8-29
What is your version of MATLAB and what GPU do you have? Have you installed the latest NVIDIA driver for that device?
回答(0 个)
另请参阅
类别
在 Help Center 和 File Exchange 中查找有关 Image Data Workflows 的更多信息
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!