PyTorch简介、发展与优点

PyTorch简介

  PyTorch是2017年1月FAIR(Facebook AI Research)发布的一款深度学习框架。从名称可以看出,PyTorch是由Py和Torch构成的。其中,Torch是纽约大学在2012年发布的一款机器学习框架,采用Lua语言为接口,但因Lua语言较为小众,导致Torch知名度不高。PyTorch是在Torch基础上用 python语言进行封装和重构打造而成的。

PyTorch发展

  • 2017年1月正式发布PyTorch
  • 2018年4月更新0.4.0版,支持Windows系统,caffe2正式并入PyTorch
  • 2018年11月更新1.0稳定版,已GitHub 增长第二快的开源项目
  • 2019年5月更新1.1.0版,支持TensorBoard,增强可视化功能
  • 2019年8月更新1.2.0版,更新torchvision,torchaudio 和torchtext,增加更多功能

    arXiv是学术界风向标。上图是2014年10月至2018年02月arXiv论文中深度学习框架提及次数统计。可以看到PyTorch的增长速度与TensorFlow一致。
    在这里插入图片描述
    Github是工业界风向标。上图是2019年3月各深度学习框架在GitHub上的Start, Forks, Watchers和Contributors数量对比。

PyTorch优点

  • 上手快:掌握Numpy和基本深度学习概念即可上手
  • 代码简洁灵活:用nn.module封装使网络搭建更方便;基于动态图机制,更灵活
  • Debug方便:调试PyTorch就像调试 Python 代码一样简单
  • 文档规范:https://pytorch.org/docs/stable/index.html可查各版本文档
  • 资源多:arXiv中的新算法大多有PyTorch实现
  • 开发者多:GitHub上贡献者(Contributors)已超过1100+
  • 背靠大树:FaceBook维护开发

适合人群

  • 深度学习初学者:模型算法实现容易,加深深度学习概念认识
  • 机器学习爱好者:数十行代码便可实现人脸识别,目标检测,图像生成等有趣实验
  • 算法研究员:最新arXiv论文算法快速复现
  • 2
    点赞
  • 12
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
PyTorch是一个开源的机器学习框架,可用于训练人工神经网络。与其他框架相比,PyTorch具有使用简便、动态计算图、灵活性高等优点。在深度学习的发展中,PyTorch已经成为了非常流行的框架之一。随着神经网络的复杂性提升,需要更有能力的硬件来加速模型训练。GPU便是其中的一种选择。GPU的并行计算能力和大量显存都可以用来提升训练速度。PyTorch天生就具备对GPU计算的支持,便于用户使用GPU进行加速。 在PyTorch中,使用GPU的方式十分简单。首先要保证PyTorch已经安装CUDA依赖,即NVidia官方提供的GPU加速计算工具包。其次,在代码中使用torch.device()指定计算设备为GPU即可。例如:device = torch.device("cuda:0")。这里的cuda:0表示第一块GPU,如果有多块GPU的话,可以选择相应的编号。有了PyTorch的GPU支持,用户只需改动少量代码,即可充分发挥GPU的加速效果,大幅缩减训练时间。 需要注意的是,GPU并不一定适合所有模型。并行计算是GPU的优势,但也需要充分发挥才能体现。如果网络过于简单,则可能得不到充分的加速效果;如果模型运算量不大,反而可能浪费显存资源。因此,用户在选择是否使用GPU时,需要综合考虑模型规模、数据大小、GPU显存等因素,才能做出最优的选择。 总之,PyTorch的GPU支持为模型训练提供了很大的便利,有助于缩短训练时间、提高训练效率,在各类深度学习任务中都具有很好的应用前景。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值