背景:
因为训练模型时,GPU利用率一直上不去,在1% -5%,通过下面几行代码,分析出
gc.collect()
耗时占了67%,注释掉GPU利用率,涨到了30%,感谢这个文章,特此转载
原文:https://blog.csdn.net/rookie_is_me/article/details/88866010
import line_profiler
def prime_num(max_num):
for num in range(2, max_num):
if num < 2:
pass
elif num == 2:
pass
else:
for i in range(2, num):
if num % i == 0:
break
else:
pass
#这将要运行的函数,放在中间
profile = line_profiler.LineProfiler(prime_num) # 把函数传递到性能分析器
profile.enable() # 开始分析
prime_num(3000) # 要执行的函数放在这里,代码执行完自会打印耗时的信息
profile.disable() # 停止分析
profile.print_stats() # 打印出性能分析结果