[英]Matlab - CuDNN Error with MatConvNet
我正在尝试使用 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);
你知道如何解决这个问题吗?
非常感谢您提前。
最有可能的是,您的 CPU 设置仍在运行。 如果您已将其更改为 GPU,则网络的图像和参数必须为 gpuArray 类型。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.