Pytorch学习-(1)
1. pytorch简介
PyTorch是一个开源的Python机器学习库,基于Torch,用于自然语言处理等应用程序。Pytorch和Numpy库很像,但是Pytorch确可以简化很多深度学习中的步骤,只有你自己下去使用一次才能体会到它的强大。
2. 安装
(1)进入Pytorch官网:link
(2)刚进去就会有一个GetStart的英文,点击进去。
(3)往下翻一点会出现以下界面:
(4) 选择自己的的配置和CUDA版本,如果你不知道怎么看CUDA版本,可以点击这个链接:link
(5)用win+r打卡cmd,之后用第4步中的Command来安装你的Pytorch,注意这个只适用于python3版本的。
3. 初步探索
3.1 数据类型
Pytorch中有两种类型的抽象数据,被称为张良和变量。张量类似于numpy中的数组,它们可以在GPU上使用,并且可以改善性能。**只有在数据被表示成数字的张量时,机器学习算法才能够理解不同格式的数据。**张量可以改变大小,在深度学习中一般有最多5维的张量。
(1)2维张量举例:
从sklearn库导入自带的波斯顿房价数据:
(2)3维张量举例:显示单通道的图片,用切片显示熊猫的脸部:
3.2 第一个神经网络
代码如下:
这是一个框架,里面各个函数的代码如下:需要导入的包:
结果如下: