pytorch实现简单的线性回归cpu版本和gpu版本

pytorch实现简单的线性回归cpu版本和gpu版本

实现步骤

  1. 准备数据
  2. 定义模型
  3. 实例化模型,实例优化器类,实例loss
  4. 循环进行梯度下降,参数更新

cpu版本实现

import torch
import torch.nn as nn
import matplotlib.pyplot as plt
from torch.optim import SGD

# 1. 准备数据
x = torch.rand([500

你可能感兴趣的:(笔记记录,pytorch,python,pytorch,线性回归,人工智能)