欢迎来到 PyTorch 入门指南!本教程将带你了解如何使用 PyTorch 搭建第一个神经网络,适合零基础学习者。📚

🧠 什么是 PyTorch?

PyTorch 是一个开源的深度学习框架,由 Facebook 的 AI 研究实验室开发。它以动态计算图(Dynamic Computation Graph)著称,非常适合研究和实验场景。
如需深入了解 PyTorch 的核心概念,可访问 PyTorch 基础教程

🛠 安装 PyTorch

  1. 使用 pip 安装
    pip install torch torchvision torchaudio
    
  2. 使用 Conda 安装
    conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
    

💡 安装完成后,建议运行以下代码验证环境:

import torch
print(torch.__version__)

🧩 快速入门示例

以下代码展示如何用 PyTorch 搭建一个简单的线性回归模型:

import torch
from torch import nn

# 定义模型
model = nn.Linear(1, 1)

# 输入数据
inputs = torch.tensor([[2.0], [3.0], [4.0]])
targets = torch.tensor([[1.0], [0.5], [-1.0]])

# 计算输出
outputs = model(inputs)
print("模型输出:", outputs)

📊 若想查看更详细的训练流程图,可点击 PyTorch 训练流程图

📚 学习资源推荐

PyTorch_Logo
神经网络结构