python代码运行过程中输出内存占用情况

python运行过程中输出内存占用情况

import os
import psutil
pid=os.getpid()#获取当前进程的id，相当于ps aux|grep train.py看到的那个进程id
process=psutil.Proceess(pid)#创建Process对象
mem_info=process.memory_info()#获取当前进程的占用情况
print('此时内存占用(RSS常驻内存集):',mem_info.rss)
print('此时内存占用(VMS虚拟内存集):',mem_info.vms)

单位是字节Byte，可以转换成GB: /1024**3