PyTorch 1.7发布,支持CUDA 11、Windows分布式训练
机器之心报道
昨日,PyTorch 团队发布 PyTorch 1.7 版本。该版本增添了很多新特性,如支持 CUDA 11、Windows 分布式训练、增加了支持快速傅里叶变换(FFT)的新型 API 等。
CUDA 11 得到 PyTorch 官方支持;
对 autograd 分析器,更新和添加了 RPC、TorchScript 和堆栈跟踪(Stack trace)的分析和性能;
(测试版)通过 torch.fft 支持 NumPy 兼容的 FFT 操作;
(原型版)支持英伟达 A100 GPU 和原生 TF32 格式;
(原型版)支持 Windows 系统上的分布式训练。
torchvision
(稳定版)transforms 支持张量输入、批处理计算、GPU 和 TorchScript
(稳定版)JPEG 和 PNG 格式的原生图像 I/O
(测试版)新型视频读取器 API
torchaudio
(稳定版)增加了对语音录制(wav2letter)、文本转语音(WaveRNN)和源分离(ConvTasNet)的支持
添加机器之心小助手:syncedaai5,邀请加入DJL交流群。
点击阅读原文,注册直播。
© THE END
转载请联系本公众号获得授权
投稿或寻求报道:content@jiqizhixin.com
关注公众号:拾黑(shiheibook)了解更多
[广告]赞助链接:
四季很好,只要有你,文娱排行榜:https://www.yaopaiming.com/
让资讯触达的更精准有趣:https://www.0xu.cn/

随时掌握互联网精彩
赞助链接
排名
热点
搜索指数
- 1 铁肩担道义 历史鉴未来 7904400
- 2 中方对会谈结果满意吗?外交部回应 7808099
- 3 中美双方降低超100%关税 7712520
- 4 中国经济必将破浪前行 7619125
- 5 汶川地震67只搜救犬已全部离世 7523481
- 6 中概股美股开盘大涨 7428153
- 7 利用工作之便出售公民个人信息?罚 7334282
- 8 纳指高开4% 美股科技七巨头集体大涨 7236905
- 9 中美各取消91%关税 暂停24%关税 7137714
- 10 黄晓明金世佳进博士复试 7047435