栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

PyTorch学习Task01 认知与安装

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

PyTorch学习Task01 认知与安装

PyTorch认知与安装
  • PyTorch简介
  • 安装
  • 了解PyTorch基本步骤

PyTorch简介
  • 深度学习框架
  • 优点:简洁;可以逐行执行
安装
  1. conda env list查看虚拟环境

    看到虚拟环境已经创建好了。
  2. 一系列下载操作,这里下载的是gpu版本。
    先查看CUDA版本,NVIDIA 控制面板 - 系统信息 - 组件

    然后去官网找对应的pytorch版本。
  3. 检查是否安装好
    conda activate pytorch_gpu进虚拟环境,打开python
import torch
torch.cuda.is_available()


返回true!就是装好了

了解PyTorch基本步骤

Pytorch编写代码基本步骤思想!

  1. 输入处理模块 (X 输入数据,变成网络能够处理的Tensor类型)
    pytorch为我们提供了Dataset和Dataloader两个类来方便的构建。

  2. 模型构建
    所有的模型都需要继承torch.nn.Module

  3. 定义代价函数和优化器

  4. 构建训练过程

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/316612.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号