目录
1、PyTorch框架介绍
Torch其实跟Tensorflow中的Tensor是一个意思,当做是能在GPU中计算的矩阵就可以啦。
意思就是现在我们有了一批数据,无论你是图像数据还是文本数据,我们都需要把数据转化成一个矩阵,接下来在建模过程当中,我们就是要对这些数据做各种各样的变换,这一些流程做完之后,得到我们想要达到的结果。PYtorch其实做了这样一些事情,他把我们所有矩阵需要计算的东西,统统的传入到GPU当中,因为GPU当中做矩阵运算比较快,GPU当中帮我们实现了所有的计算功能,整体的一个计算,从前向传播到反向传播,有可能涉及到的非常复杂的计算,这些计算统统框架都帮我们实现了,我们要去做的就是去设计整个的网络流程,整个的架构就可以了,所以说深度学习框架说白了就是一个计算的工具,帮我们实现由前到后整体的计算过程。
2、安装Pytorch
官网:PyTorch
使用pip的安装方法比较简单
2.1、CPU版本的安装命令:
pip3 install torch torchvis