PyTorch 1.8 已发布,此版本由 1.7 发布以来的 3000 多个 commit 组成,重要更新内容包括在编译、代码优化和科学计算前端 API 方面的优化和改进,以及通过 pytorch.org 提供的二进制文件来支持 AMD ROCm。1.8 版本还为大规模训练管道、模型并行化,以及梯度压缩提供了特性改进。部分值得关注的亮点: 支持通过torch.fx进... PyTorch 1.8 发布:更新编译器和分布式训练、为移动端引入新教程下载地址