pytorch实现简单的线性回归cpu版本和gpu版本 实现步骤 准备数据 定义模型 实例化模型,实例优化器类,实例loss 循环进行梯度下降,参数更新 cpu版本实现 import torch import torch.nn as nn import matplotlib.pyplot as plt from torch.optim import SGD # 1. 准备数据 x = torch.rand([500