0

我目前正在尝试使用 gpu 使用 cnn 在 matlab 中训练和测试网络。我有 gpu -GTX 1070 和 CUDA 8.0 和 cudnn 5.1。我从链接https://www.mathworks.com/help/vision/examples/object-detection-using-deep-learning.html遵循了使用 mathworks 提供的深度学习进行对象检测的示例 在运行以下几行之后：

        % Train a network.
        Net1 = trainNetwork(trainingImages, trainingLabels, layers, opts);

它给了我训练有素的网络作为输出。但是在我运行以下行之后

        YTest = predict(Net1, X_test);

它给了我以下错误

    Error using nnet.internal.cnngpu.convolveForward2D
Unexpected error calling cuDNN: CUDNN_STATUS_BAD_PARAM.

Error in nnet.internal.cnn.layer.Convolution2D/doForward (line 218)
            Z = nnet.internal.cnngpu.convolveForward2D( ...

Error in nnet.internal.cnn.layer.Convolution2D/forwardNormal (line 195)
            Z = this.doForward(X,this.Weights.Value,this.Bias.Value);

Error in SeriesNetwork/predict (line 139)
                Y(:,:,:,i) = predictNetwork.predict(X);
4

1 回答 1

1

对于 Pascal 卡，您需要获取 cuDNN 补丁： https ://www.mathworks.com/support/bugreports/1439741

于 2017-01-18T13:17:11.273 回答