Matlab - MatConvNet 的 CuDNN 错误答案

【问题标题】：Matlab - CuDNN Error with MatConvNetMatlab - MatConvNet 的 CuDNN 错误
【发布时间】：2021-01-24 10:18:43
【问题描述】：

我正在尝试使用 Matlab 和 Matconvnet 进行深度学习，尤其是在这个实现上工作：Finding Tiny Faces。

我正在使用 Cuda 9 (GTX 1080Ti) 开发 Ubuntu 16.04。对于实现，我使用的是 Matlab R2017b。我已经按照说明安装和编译 Matconvnet vl_compilenn('enableImreadJpeg', true) 并且我已经通过了提供的测试。

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

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);

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

【问题讨论】：

标签： matlab ubuntu deep-learning matconvnet

【解决方案1】：

很可能，您的 CPU 设置仍处于开启状态。如果您已将其更改为 GPU，则网络的图像和参数需要为 gpuArray 类型。

【讨论】：