Pytorch实战 | 第P2周：彩色图片识别

• 难度：新手入门⭐
• 语言：Python3、Pytorch

🍺 要求：

1. 学习如何编写一个完整的深度学习程序
2. 手动推导卷积层与池化层的计算过程

🏡 我的环境：

• 语言环境：Python3.8
• 编译器：jupyter notebook
• 深度学习环境：Pytorch

一、 前期准备

1. 设置GPU

如果设备上支持GPU就使用GPU,否则使用CPU

import torch
import torch.nn as nn
import matplotlib.pyplot