MATLAB-与MatConvnet的Cudnn误差

Question

我正在尝试使用Matlab和Matconvnet进行深度学习，尤其是从事此实现：寻找微小的面孔。

我正在使用ubuntu 16.04，使用CUDA 9（GTX 1080TI）。对于实施，我正在使用MATLAB R2017B。我已经按照说明进行安装和编译Matconvnet vl_compilenn('enableImreadJpeg', true)，并且通过了家具测试。

当我尝试运行第一个函数bboxes = tiny_face_detector('data/demo/selfie.jpg', './selfie.png', 0.5, 0.1, 1)时，我有一个错误：

Error using vl_nnconv
forward: cuDNN error [cudnn:
"/home/alexattia/Work/RecVis/tiny/matconvnet/matlab/src/bits/impl/nnconv_cudnn.cu":141
(CUDNN_STATUS_BAD_PARAM)]
Error in dagnn.Conv/forward (line 11)
      outputs{1} = vl_nnconv(...
Error in dagnn.Layer/forwardAdvanced (line 85)
      outputs = obj.forward(inputs, {net.params(par).value}) ;
Error in dagnn.DagNN/eval (line 91)
  obj.layers(l).block.forwardAdvanced(obj.layers(l)) ;
Error in tiny_face_detector (line 130)
  net.eval(inputs);

您知道如何解决这个问题吗？
非常感谢您。

Answer 1

很可能您的CPU设置仍在打开。如果将其更改为GPU，则网络的图像和参数需要为GPUArray类型。