windows-matlab环境下,生成的caffe模型无法运行

我的环境如下:windows10,matlab2014a
在配置完MATLAB的接口之后,可以运行MATLAB自带的测试程序:
[scores, maxlabel] = classification_demo(im, use_gpu)
然后我准备了mnist例程的相关文件,如下:
deploy.prototxt和lenet_iter_10000.caffemodel
我试图输入一张图片,然后运行得到分类结果,但运行的结果是matlab卡死,程序自动退出,不知道哪里出了问题,代码如下:
if exist('../+caffe', 'dir')
  addpath('..');
else
  error('Please run this demo from caffe/matlab/demo');
end

% Set caffe mode
if exist('use_gpu', 'var') && use_gpu
  caffe.set_mode_gpu();
  gpu_id = 0;  % we will use the first gpu in this demo
  caffe.set_device(gpu_id);
else
  caffe.set_mode_cpu();
end

% Initialize the network using BVLC CaffeNet for image classification
% Weights (parameter) file needs to be downloaded from Model Zoo.
model_dir = '../../models/mnist/';
net_model = [model_dir 'deploy.prototxt'];
net_weights = [model_dir 'lenet_iter_10000.caffemodel'];
phase = 'test'; % run with phase test (so that dropout isn't applied)
if ~exist(net_weights, 'file')
  error('Please download CaffeNet from Model Zoo before you run this demo');
end

% Initialize a network
net = caffe.Net(net_model, net_weights, phase);
 
每次都是运行到net = caffe.Net(net_model, net_weights, phase);就卡死 了
高手请指教
已邀请:

xfwaaang

赞同来自: caffe

已解决
运行saveFilters.m(生成x3.mat),其实工作是: *.caffemodel -> *.mat ,MATLAB 强行停止
原因:cuda与cudnn版本不兼容
1).更新NVIDIA驱动,去官网下载对应版本
2).重新安装cuda与cudnn

我是如何发现问题原因并解决的?
因为直接在MATLAB中运行 saveFilters.m ,总是直接卡死,停止运行,且没有任何提示,所以我一直找不到解决办法。
我试着将 saveFilters.m 转化成 saveFilters.exe ,双击运行后,弹出窗口显示了生成网络的log,最后有个错误提示:
Check failed: status == CUDNN_STATUS_SUCCESS (1 vs. 0)  CUDNN_STATUS_NOT_INITIALIZED

若卜黎

赞同来自:

同样出现这个问题,运行classification_demo就卡死退出。

xfwaaang

赞同来自:

请问你的问题解决了没?
我也出现了同样的问题
 

要回复问题请先登录注册