PyTorch 1.7 发布:支持 CUDA 11、FFT 新 API、及 Windows 分布式训练

栏目: 软件资讯 · 发布时间: 4年前

内容简介:PyTorch 1.7 现已发布,此版本增加了许多新的 API、包括对与 NumPy 兼容的 FFT 操作的支持、用于性能分析的工具、以及对基于分布式数据并行(Distributed Data Parallel,DDP)和基于远程过程调用(Remote Proced...

PyTorch 1.7 现已发布,此版本增加了许多新的 API、包括对与 NumPy 兼容的 FFT 操作的支持、用于性能分析的 工具 、以及对基于分布式数据并行(Distributed Data Parallel,DDP)和基于远程过程调用(Remote Procedure Call, RPC)的分布式训练的重大更新。

此外,一些功能已经稳定下来,包括自定义 C++ 类、内存分析器、通过类似自定义张量对象的扩展、RPC 中的用户异步函数、以及 torch 中的许多其他功能,例如 Per-RPC timeout、DDP dynamic bucketing 和 RRef helper。

一些更新亮点如下:

  • 正式支持 CUDA 11 ,二进制文件可通过 PyTorch.org下载。
  • 在 autograd 分析器中更新和添加 RPC、 TorchScript 和 Stack traces 的分析和性能
  • (Beta)通过 torch.fft 支持兼容 NumPy 的快速傅里叶变换(FFT)
  • (原型)支持 Nvidia 新一代 A100 GPU 和原生 TF32 格式
  • (原型)现在支持 Windows 上的分布式训练
  • torchvision
    • (稳定) 变换现在支持 Tensor 输入、批量计算、GPU 和TorchScript
    • (稳定)用于 JPEG 和 PNG 格式的原生图像 I/O
    • (Beta)新的视频读取器 API
  • torchaudio
    • (稳定)增加了对语音记录(wav2letter)、文本到语音(WaveRNN)和源分离(ConvTasNet)的支持

值得注意的是,从 PyTorch 1.6 开始,功能的状态将分为三种,分别是稳定、Beta 和原型。

可在此处找到完整的发行说明。


以上就是本文的全部内容,希望本文的内容对大家的学习或者工作能带来一定的帮助,也希望大家多多支持 码农网

查看所有标签

猜你喜欢:

本站部分资源来源于网络,本站转载出于传递更多信息之目的,版权归原作者或者来源机构所有,如转载稿涉及版权问题,请联系我们

逻辑的引擎

逻辑的引擎

[美] 马丁·戴维斯 / 张卜天 / 湖南科学技术出版社 / 2005-5 / 20.00元

本书介绍了现代计算机背后的那些基本概念和发展这些概念的人,描写了莱布尼茨、布尔、费雷格、康托尔、希尔伯特、哥德尔、图灵等天才的生活和工作,讲述了数学家们如何在成果付诸应用之前很久就已经提出了其背后的思想。博达著作权代理有限公司授权出版据美国W.W.Norton公司2000年版本译出。2007年第二版亦使用同一ISBN。一起来看看 《逻辑的引擎》 这本书的介绍吧!

HTML 编码/解码
HTML 编码/解码

HTML 编码/解码

RGB CMYK 转换工具
RGB CMYK 转换工具

RGB CMYK 互转工具

HEX CMYK 转换工具
HEX CMYK 转换工具

HEX CMYK 互转工具