【PyTorch】2.0 入门学习
- [1. 搭建PyTorch2.0环境](#1. 搭建PyTorch2.0环境)
1. 搭建PyTorch2.0环境
PyTorch是一个用于机器学习和深度学习的开源深度学习框架,由Facebook于2016年发布,其主要实现了自动微分功能,并引入动态计算图使模型建立更加灵活。Pytorch可分为前后端两个部分,前端是与用户直接交互的python API,后端是框架内部实现的部分,包括Autograd,它是一个自动微分引擎。
为什么使用PyTorch2.0
- 训练速度快:
compiled_model = torch.compile(model)把你写的"动态Python代码"自动编译优化成高性能计算图,训练/推理速度提升(20%~2倍,甚至更高);简单总结一句话:👉 PyTorch 2.0 = "写代码不变 + 自动帮你加速" - 可用性强:从早期版本开始,API一直保持一致,支持CPU、GPU、并行处理以及分布式训练
- 100%向后兼容