Как изменить движок caffe на CAFFE вместо CUDNN, не перестраивая caffe и не изменяя файлы .prototxt - PullRequest
0 голосов
/ 09 апреля 2020

Я получаю ошибку: cudnn_relu_layer.cpp:13] Check failed: status == CUDNN_STATUS_SUCCESS (4 vs. 0) CUDNN_STATUS_INTERNAL_ERROR Я искал причину для inte rnet, и я понял, что Caffe не может использовать cudnn. Я использую Cudnn версии 7.1.1 и cuda версии 9.0, и Cudnn включен в Makefile.config caffe. На моем компьютере есть другие программы, которые зависят от этих версий cuda и cudnn, поэтому я не хочу их менять. Я посмотрел на deploy.prototxt модели и не нашел ключевое слово 'engine', но нашел его на caffe.proto. Итак, мой вопрос: как мне установить caffe так, чтобы все слои имели в качестве движка cafe, а не cudnn?

...