Caffe | Echec de la vérification: erreur == cudaSuccess (2 vs. 0) insuffisante en mémoire
J'essaie de former un réseau sur Caffe. J'ai une taille d'image de 512x640. La taille du lot est 1. J'essaie d'implémenter FCN-8s .
J'exécute actuellement cette application sur une instance Amazon EC2 (g2.2xlarge) avec 4 Go de mémoire GPU. Mais quand je lance le solveur, il jette immédiatement une erreur
Check failed: error == cudaSuccess (2 vs. 0) out of memory *** Check failure stack trace: *** Aborted (core dumped)
Quelqu'un peut-il m'aider à partir d'ici?
L'erreur que vous obtenez est en effet à court de mémoire, mais ce n'est pas la RAM, mais plutôt la mémoire GPU (notez que l'erreur provient de CUDA).
Habituellement, lorsque Caffe manque de mémoire - la première chose à faire est de réduire la taille du lot (au prix de la précision du gradient ), mais puisque vous êtes déjà à la taille du lot = 1 ...
Êtes-vous sûr que la taille du lot est 1 pour les phases les deux TRAIN et TEST?
Caffe peut utiliser plusieurs GPU. Ceci est seulement supporté dans l'interface C++, pas dans l'interface python. Vous pouvez également activer cuDNN pour réduire l'encombrement de la mémoire.