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 习近平文化思想在浙江的溯源与实践 7904124
- 2 救下一家三口后牺牲的小伙下葬 7809041
- 3 山西载12人中巴失联 已发现1具遗体 7713120
- 4 本轮降雨为何如此强盛 7619449
- 5 佛教协会:对释永信戒牒予以注销 7520963
- 6 2025百度热搜年中报告出炉 7427923
- 7 没考上 高校发“未录取通知书” 7329530
- 8 汪峰罕见带4娃同行 7232341
- 9 揭“大熊猫被虐”谣言背后利益链 7143139
- 10 司机与13岁“巴士迷”车内发生关系 7042928