PyTorch框架:(1)基本处理操作

本文是PyTorch框架的基础教程,介绍了PyTorch的安装、基本使用方法,特别是自动求导机制。通过实例展示了如何进行线性回归的CPU和GPU训练。还探讨了tensor格式和PyTorch的hub模块,提供了调用预训练模型的便捷方式。
摘要由CSDN通过智能技术生成

目录

1、PyTorch框架介绍

2、安装Pytorch

2.1、CPU版本的安装命令:

2.2、GPU版本的安装命令:

2.2.1、安装CUDA

3、基本使用方法

 4、Pytorch中的自动求导机制

4.1、例子: 

 ​

 4.2、做一个线性回归试试水

4.2.1、CPU训练版本

  4.2.1、GPU训练版本

5、常见tensor格式

 6、Pytorch中非常强大的模块:hub模块

 6.1、Pytorch hub模块官网:

  6.2、Pytorch hub 的github 站点:

 6.3、演示


1、PyTorch框架介绍

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

2、安装Pytorch

官网:PyTorch

使用pip的安装方法比较简单

2.1、CPU版本的安装命令:

pip3 install torch torchvis
  • 1
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Upupup6

写手不易请留下你的打赏鼓励谢谢

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值