PyTorch学习Task01 认知与安装

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. 构建训练过程