使用FP8加速PyTorch训练的两种方法总结

在PyTorch中,FP8(8-bit 浮点数)是一个较新的数据类型,用于实现高效的神经网络训练和推理。它主要被设计来降低模型运行时的内存占用,并加快计算速度,同时尽量保持训练和推理的准确性。虽然PyTorch官方在标准发布中尚未全面支持FP8,但是在2.2版本中PyTorch已经包含了对FP8的“有限支持”并且出现了2个新的变量类型,

  1. torch.float8_e4m3fn

  1. torch.float8_e5m2

,而H100也支持这种类型,所以这篇文章我们就来介绍如何使用FP8来提高训练效率

https://avoid.overfit.cn/post/0dd1fba546674b48b932260fa8742971

posted @   deephub  阅读(153)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 全程不用写代码,我用AI程序员写了一个飞机大战
· DeepSeek 开源周回顾「GitHub 热点速览」
· 记一次.NET内存居高不下排查解决与启示
· MongoDB 8.0这个新功能碉堡了,比商业数据库还牛
· 白话解读 Dapr 1.15:你的「微服务管家」又秀新绝活了
历史上的今天:
2023-05-24 Jupyter Notebook 10个提升体验的高级技巧
2022-05-24 使用动图深入解释微软的Swin Transformer
点击右上角即可分享
微信分享提示