优化以下代码把检测到的内存容量取整数例如实际检测值接近8GB取值81924GB内存取值409616GB 1638432gb 3276864gb 65534import psutil import sys def get_memory_size mem_info = psutilvirtual_memory total_mem = mem_infototal return

import psutil import sys

def get_memory_size(): mem_info = psutil.virtual_memory() total_mem = mem_info.total return total_mem

if name == 'main': mem_size = int(get_memory_size() / 1024 / 1024) # 取整数 if len(sys.argv) > 1: if sys.argv[1] == '-mem': if len(sys.argv) > 2: if int(sys.argv[2]) != mem_size: print('内存容量不足，实际容量为：', mem_size, 'MB') else: print('请指定内存容量') else: print('无效参数') else: print('请指定参数'