solidot新版网站常见问题,请点击这里查看。

PyTorch 2.0 释出

人工智能
Wilson (42865)发表于 2023年03月16日 15时05分 星期四

来自哈特拉斯船长历险记
PyTorch 基金会宣布释出 PyTorch 2.0。PyTorch 是一个开源 Python 机器学习库。新版的变化包括:稳定版 Accelerated Transformers(前称 Better Transformers),torch/compile 作为其主 API,Nvidia 和 AMD GPU 将依靠 OpenAI Triton 深度学习编译器生成高性能代码隐藏底层硬件细节,Metal Performance Shaders 后端在 macOS 平台提供了 GPU 加速的 PyTorch,在 AWS Graviton CPU 上提供更快的推理性能,等等。更多可浏览发布公告。


https://pytorch.org/blog/pytorch-2.0-release/