- PyTorch简介
- 安装
- 了解PyTorch基本步骤
- 深度学习框架
- 优点:简洁;可以逐行执行
- conda env list查看虚拟环境
看到虚拟环境已经创建好了。 - 一系列下载操作,这里下载的是gpu版本。
先查看CUDA版本,NVIDIA 控制面板 - 系统信息 - 组件
然后去官网找对应的pytorch版本。 - 检查是否安装好
conda activate pytorch_gpu进虚拟环境,打开python
import torch torch.cuda.is_available()
返回true!就是装好了
Pytorch编写代码基本步骤思想!
-
输入处理模块 (X 输入数据,变成网络能够处理的Tensor类型)
pytorch为我们提供了Dataset和Dataloader两个类来方便的构建。 -
模型构建
所有的模型都需要继承torch.nn.Module
-
定义代价函数和优化器
-
构建训练过程



