ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

PyTorch Task1 : PyTorch的认知与安装

2021-10-12 12:31:44  阅读:246  来源: 互联网

标签:Task1 步骤 认知 pytorch Pytorch PyTorch conda 安装


目录

Pytorch的认知

Pytorch的优势

1.用户增长快速各深度学习平台使用比例图
2.提供python接口
3 相比与torch增加了自动求导
4 相比与tf的图模型实现了一种动态的构建方式,类似chainer 和DyNet
5.兼容numpy

Pytorch编写代码基本步骤思想

在我们要用pytorch构建自己的深度学习模型的时候,基本上都是下面这个流程步骤,写在这里让一些新手童鞋学习的时候有一个大局感觉,无论是从自己写,还是阅读他人代码,按照这个步骤思想:
默念4大步骤,
1.找数据定义、
2.找model定义、
3.(找损失函数、优化器定义),
4.主循环代码逻辑,直接去找对应的代码块,会简单很多。

Pytorch编写代码基本步骤思想

在这里插入图片描述
具体分为四大步骤:

1、输入处理模块 (X 输入数据,变成网络能够处理的Tensor类型)

2、模型构建模块 (主要负责从输入的数据,得到预测的y^, 这就是我们经常说的前向过程)

3、定义代价函数和优化器模块 (注意,前向过程只会得到模型预测的结果,并不会自动求导和更新,是由这个模块进行处理)

4、构建训练过程 (迭代训练过程,就是上图表情包的训练迭代过程)

PyTorch的安装(Win10+GTX1060+Anaconda3)

  1. 更新显卡驱动(以更新cuda版本)
    英伟达驱动官网
  2. 查看CUDA版本:NVIDIA 控制面板 =>帮助=> 系统信息 => 组件,一般cuda版本>10,否则会限制pytorch版本。
    在这里插入图片描述
  3. 从pytorch官网获取安装命令 https://pytorch.org/
    在这里插入图片描述
  4. 在conda中安装pytorch
    如果网络较稳定,可选择在线安装。
    先切换至所要安装的环境,输入:
#添加国内镜像源
conda config --add channels conda config --add channels
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

在线安装:输入安装命令

#复制pytorch官网命令
#注意去掉- pytorch,否则不能使用国内源下载
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c conda-forge

离线安装:如果因为网络问题下载不了,卡在pytorch下载页面,就到清华镜像去下载
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/,这里可以找到所有版本的pytorch
下载好安装包之后,在当前环境下conda install +路径含安装包名称后缀,即可。
离线下载

测试安装
在这里插入图片描述
若显示true则成功。

标签:Task1,步骤,认知,pytorch,Pytorch,PyTorch,conda,安装
来源: https://blog.csdn.net/weixin_44847372/article/details/120720563

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有