PyTorch 是一个流行的开源机器学习库，特别适用于深度学习和计算机视觉任务。以下是一些 PyTorch 快速入门的要点：

安装 PyTorch

首先，您需要安装 PyTorch。以下是在 Ubuntu 系统上安装 PyTorch 的步骤：

  1. 打开终端。
  2. 运行以下命令安装 PyTorch：
pip install torch torchvision

简单的 PyTorch 示例

以下是一个简单的 PyTorch 示例，演示了如何定义一个神经网络并训练它：

import torch
import torch.nn as nn
import torch.optim as optim

# 定义一个简单的神经网络
class SimpleNet(nn.Module):
    def __init__(self):
        super(SimpleNet, self).__init__()
        self.fc = nn.Linear(10, 1)

    def forward(self, x):
        return self.fc(x)

# 创建模型实例
model = SimpleNet()

# 定义损失函数和优化器
criterion = nn.MSELoss()
optimizer = optim.SGD(model.parameters(), lr=0.01)

# 模拟一些训练数据
inputs = torch.randn(64, 10)
targets = torch.randn(64, 1)

# 训练模型
for epoch in range(100):
    optimizer.zero_grad()
    outputs = model(inputs)
    loss = criterion(outputs, targets)
    loss.backward()
    optimizer.step()

    if (epoch+1) % 10 == 0:
        print(f'Epoch [{epoch+1}/100], Loss: {loss.item():.4f}')

扩展阅读

如果您想深入了解 PyTorch，可以阅读以下本站链接：

PyTorch Logo