在一个GPU上运行的多个深度学习框架会导致CUDNN_STATUS_BAD_PARAM

Question

我想在一个c ++项目中运行Caffe和TensorFlow（仅评估），但会引起标题错误。出于某些原因，我无法将Caffe模型重新训练为TensorFlow或其他方式。据我所知，这个错误意味着GPU很忙，只为一个框架提供服务。我能做什么？

我有两个初始化函数来加载Caffe和TF模型以及两个预测函数。

Answer 1

好的，我已经解决了。我的Caffe和TF都使用cuDNN，这似乎只允许在一个GPU上运行一个框架（或实例？）。我只是关掉了Caffe的cuDNN，它确实有效。