web-dev-qa-db-fra.com

Erreur: MOO lors de l'allocation du tenseur avec la forme

je rencontre un problème avec mon modèle de démarrage lors des tests de performances avec Apache JMeter.

Erreur: MOO lors de l'allocation du tenseur avec la forme [800,1280,3] et le type float on/job: localhost/replica: 0/task: 0/device: GPU: 0 by allocator GPU_0_bfc [[Node: Cast = CastDstT = DT_FLOAT, SrcT = DT_UINT8, _device = "/ job: localhost/replica: 0/task: 0/device: GPU: 0"]] Astuce: Si vous souhaitez voir une liste des tenseurs alloués lorsque le MOO se produit, ajoutez report_tensor_allocations_upon_oom à RunOptions pour le courant informations d'allocation.

7
Mohd Daoud

Si vous avez plusieurs GPUS à portée de main, veuillez sélectionner un GPU qui n'est pas aussi occupé que celui-ci (raisons possibles, d'autres processus s'exécutent également sur ce GPU). Accédez au terminal et saisissez

export CUDA_VISIBLE_DEVICES=1 

où 1 est le nombre d'autres GPU disponibles. réexécutez le même code.

vous pouvez vérifier les GPU disponibles en utilisant

nvidia-smi 

cela vous montrera quels GPU sont disponibles et combien de mémoire est disponible sur chacun d'eux

0
Priyank Pathak