Matlab - MatConvNet 的 CuDNN 錯誤

Question

我正在嘗試使用 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);

你知道如何解決這個問題嗎？
非常感謝您提前。

Answer 1

最有可能的是，您的 CPU 設置仍在運行。 如果您已將其更改為 GPU，則網絡的圖像和參數必須為 gpuArray 類型。