Comment garder trace de la mémoire virtuelle utilisée lors de l'exécution de code Python?

Question

J'ai un paquet Python que je suis benchmarking pour la mémoire virtuelle utilisée.

À l'heure actuelle, ce n'est pas très précis. Je soumets le script par exemple

python script1.py 

et de regarder le temps et les ressources utilisées pour exécuter.

Existe-t-il un package/une méthode permettant de savoir où se trouvent mes goulots d'étranglement? Je pense à un outil comme

ram_used python script1.py 

qui garderait une trace de la quantité de RAM utilisée.

Une telle chose existe-t-elle?

Answer 1

Oui, il y a - Ça s'appelle memory_profiler. Une fois que vous avez terminé et ajouté le décorateur @profile aux fonctions que vous souhaitez profiler, vous pouvez utiliser;

python -m memory_profiler example.py 

Answer 2

Il y a aussi memory_infopsutil. https://pythonhosted.org/psutil/

place au début du script:

import psutil 
p = psutil.Process() 

et étant donné que le code fonctionne correctement, placez ceci sur la dernière ligne:

print(p.memory_info())