用微信扫描二维码

大模型动力引擎:PyTorch性能与显存优化手册

大模型动力引擎:PyTorch性能与显存优化手册

作者:张爱玲

2024.10.1 出版

可语音朗读

开通电子书VIP
62.30得到贝

主编推荐语

助你理解、掌握PyTorch的优化技巧。

内容简介

本书致力于探索如何在大规模深度学习模型训练中,最大限度地提高性能和优化显存使用。本书面向深度学习从业者,尤其是希望深入了解并提升模型训练效率的工程师与研究人员。随着深度学习模型和数据规模的迅速增长,如何高效利用硬件资源,减少训练时间,成为当前AI系统工程的关键挑战。 本书内容涵盖从单机到分布式训练,从显存管理到性能分析的多种优化策略,力求通过丰富的代码实例和深入的原理讲解,使读者能够在实践中灵活应用这些方法。

出版方

清华大学出版社