使用memory_profiler时无法导入模块

Posted 2023-01-06

我正在尝试使用memory_profiler，但是遇到了example.py中孤立的问题：

#! /usr/bin/env python 
import argparse

def parse_args():
    parser = argparse.ArgumentParser(description='Dummy description')
    parser.add_argument('--option',  action='store_true')
    return parser.parse_args() 

if __name__ == '__main__':
    parse_args()

所以基本上只使用argparse。独立运行良好（我有Python3.3）。但是，当我发出

$ python -m memory_profiler example.py

我得到一个错误：

NameError: global name 'argparse' is not defined

此外，如果我把它放在电话里

    parser = argparse.ArgumentParser(description='Dummy description')

下方

    if __name__ == '__main__':

并且我注释掉对parse_args（）的函数调用，然后我没有收到错误。

有人知道这里出了什么问题吗？

答案

根据我的经验，内存分析器未在其PYTHONPATH中使用当前目录，所以我的解决方法是使用以下命令：

PYTHONPATH=./<CORRECT_PATH> python -m memory_profiler example.py