caffe 运行mnist数据集 无论cpu还是gpu运行都出现CUDNN_STATUS_NOT_SUPPORTED

开发环境为 windows10 + cuda9.0 + cudnn7.3 显卡为GTX960M(笔记本显卡).在已经通过微软版本编译好了caffe以后。第一次运行mnist数据集的时候。出现问题
 
mnist配置文件没有改动...除了图片数据的src调整了一下
 
这是我的两个配置文件
lenet_solve.prototxt和lenet_train_test.prototxt
 
 # The train/test net protocol buffer definition
net: "examples/mnist/lenet_train_test.prototxt"
# test_iter specifies how many forward passes the test should carry out.
# In the case of MNIST, we have test batch size 100 and 100 test iterations,
# covering the full 10,000 testing images.
test_iter: 100
# Carry out testing every 500 training iterations.
test_interval: 500
# The base learning rate, momentum and the weight decay of the network.
base_lr: 0.01
momentum: 0.9
weight_decay: 0.0005
# The learning rate policy
lr_policy: "inv"
gamma: 0.0001
power: 0.75
# Display every 100 iterations
display: 100
# The maximum number of iterations
max_iter: 10000
# snapshot intermediate results
snapshot: 5000
snapshot_prefix: "examples/mnist/lenet"
# solver mode: CPU or GPU
solver_mode: CPU


第二个文件没改过,就不发了。
 
不论cpu还是gpu都显示这个错误
 
 
 
 
1.jpg
已邀请:

hlm0830

赞同来自:

caffe-master\windows\CommonSetting.props(没有的话,你把旁边的那个后缀还有.examples的复制一份,改成我这个名字)里面有写到
 
<!-- CuDNN 4 and 5 are supported -->
又因为cudnn5.0zui最多支持到cuda8.0.所以想用cuda9.0的话,需要把上面的
第八行:     <UseCuDNN>true</UseCuDNN>  中true改为false
 
而不想用的话,直接卸载9.0而接下来下载cuda8.0和cudnn5.0,第二种方式比第一种到时候用起来快。但是我觉得新版本更酷,而且我懒,所以弄了第一种。原问题已经解决,算是自问自答把。所以给大家一个提示。cudnn必须在4.0和5.0之间!!!!!!!!!!
 

要回复问题请先登录注册