作品简介
本书致力于探索如何在大规模深度学习模型训练中,最大限度地提高性能和优化显存使用。本书面向深度学习从业者,尤其是希望深入了解并提升模型训练效率的工程师与研究人员。随着深度学习模型和数据规模的迅速增长,如何高效利用硬件资源,减少训练时间,成为当前AI系统工程的关键挑战。
本书内容涵盖从单机到分布式训练,从显存管理到性能分析的多种优化策略,力求通过丰富的代码实例和深入的原理讲解,使读者能够在实践中灵活应用这些方法。
出版时间2024-10-01
共 14 章
作者简介试读
内容简介试读
前言试读
01 欢迎来到这场大模型竞赛试读
02 深度学习必备的硬件知识试读
03 深知度识学习必备的PyTorch
04 定位性能瓶颈的工具和方法
05 数据加载和预处理专题
06 单卡性能优化专题
07 单卡显存优化专题
08 分布式训练专题
09 高级优化方法专题
10 GPT-2优化全流程
结语

