热门搜索 :
考研考公
您的当前位置:首页正文

PyTorch:60分钟入门学习

来源:东饰资讯网

最近在学习PyTorch这个深度学习框架,在这里做一下整理分享给大家,有什么写的不对或者不好的地方,还请大侠们见谅啦~~~

写在前面

本文就是主要是对PyTorch的安装,以及入门学习做了记录,也可以算是用自己的理解对文档进行了翻译。

PyTorch 是什么

  • PyTorch是一个深度学习框架;
  • 支持GPU并行处理的张量计算;
  • 支持动态神经网络的构建;
  • 比较容易学习。

大多数的框架,例如常见的caffe,TensorFlow都需要构建静态图,构建好一个神经网络之后需要重复使用相同的结构。而PyTorch则允许改变网络的运行方式,在构建神经网络时只需要定义网络中的层结构以及前向传播函数即可,反向传播求梯度的过程它都会自动完成。

除此,使用起来比较方便,对于我自己这样的新学者来说真的是非常友善啦,官网提供了详细的安装以及各种入门指导,还提供了丰富的examples供我们学习~

PyTorch 的安装

PyTorch的安装在官网上有给出方法,按照


按照自己的电脑配置去选择,pip下记得要加sudo。
我自己的环境是:ubuntu14.04+anaconda
具体的配置过程戳这里

PyTorch 60分钟入门教程

安装好之后,可以来试一试了~

初识PyTorch之Tensor

  • 包导入
  • 构建矩阵
  • 构建随机矩阵
  • 获取Tensor大小
  • Tensor的4种加法
  • Tensor索引方法
  • Tensor与numpy array的相互转换
  • Tensor移到GPU上

PyTorch之自动求梯度

PyTorch之神经网络

  • 怎么样定义一个神经网络
  • 处理输入以及调用backward函数
  • 计算损失函数
  • 迭代更新网络的权重

PyTorch之训练分类器


展示的是各个类别测试过程中的精度。
基本上简单的网络都可以照着这个思路去实现出来。

PyTorch之并行计算

参考文献

陆陆续续终于整理完了这篇文章,呼一口气~

正经脸:小菜鸟一只,有问题还请多多指教!

Top