Plusieurs environnements d'apprentissage en profondeur s'exécutant sur un GPU provoque CUDNN_STATUS_BAD_PARAM

Question

Je veux exécuter Caffe et TensorFlow (évaluation seulement) dans un projet C++ mais provoque l'erreur en tant que titre mentionné. Pour certaines raisons, je ne peux pas recycler le modèle Caffe à TensorFlow ou l'inverse. A ma connaissance, cette erreur signifie que le GPU est occupé et ne fournit un service que pour un framework. Tout ce que je peux faire?

J'ai deux fonctions init pour charger le modèle Caffe et TF et deux fonctions de prédiction aussi.

Answer 1

OK, je l'ai résolu. Mon Caffe et TF utilisent tous les deux cuDNN qui semble permettre seulement un cadre (ou instance?) Fonctionnant sur un GPU. Je viens d'éteindre le cuDNN du Caffe et ça marche.