PyTorch 1.10 正式版发布 ,更新了 CUDA Graphs API

PyTorch 是一个基于 Torch 库的开源机器学习库,用于计算机视觉和自然语言处理等应用,主要由 Facebook 的 AI 研究实验室开发。 它是在修改后的 BSD 许可下发布的免费开源软件。 尽管 Python 接口更加完善并且是开发的主要重点,但 PyTorch 也有一个 C++ 接口。

PyTorch 团队今天宣布发布 PyTorch 1.10。这个期待已久的更新由自上一个版本1.9以来的 3400 多次提交组成。这个新的主要版本由 426 位贡献者完成。真诚地感谢社区不断改进 PyTorch。

PyTorch 1.10 更新侧重于改进 PyTorch 的训练和性能以及开发人员的可用性。完整的发行说明可在此处获得

亮点:

在 1.10 版中,CUDA 工作负载的 CPU 开销与集成的 CUDA Graphs API 一起考虑。在这个新版本的 PyTorch 中,FX、torch.special 和 nn.Module Parametrization 等几个前端 API 已从测试版变为稳定版。新版本还支持 JIT Compiler 中的自动融合,除了 GPU 之外还扩展到 CPU。Android NNAPI 支持现已推出测试版。

PyTorch 是任何深度学习项目的完美库。它可以轻松构建具有强大 GPU 加速的快速、可靠的网络,您可以在 Python 中轻松阅读和理解。例如,您可以考虑具有强大 GPU 加速的 Numpy。

PyTorch 1.10 是最新的更新,它将进一步改进 Pytorch 的训练和性能,并使开发人员更容易更有效地使用他们的技能。

Github:https://github.com/pytorch/pytorch/releases/tag/v1.10.0

PyTorch 博客:https://pytorch.org/blog/pytorch-1.10-released/

The post PyTorch 1.10 正式版发布 ,更新了 CUDA Graphs API first appeared on Linux迷.

版权声明:
作者:玉兰
链接:https://www.techfm.club/p/4555.html
来源:TechFM
文章版权归作者所有,未经允许请勿转载。

THE END
分享
二维码
< <上一篇
下一篇>>