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
  댓글 수: 1
Joss Knight
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 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