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都显示这个错误
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 个回复
hlm0830
赞同来自:
<!-- 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之间!!!!!!!!!!