### 如何在Kaggle平台上使用PyTorch #### 安装依赖库 由于Kaggle Notebook已经预安装了许多常用的机器学习框架，包括PyTorch，在大多数情况下不需要额外安装。如果需要特定版本的PyTorch或其他依赖项，则可以通过`!pip install`命令来安装。 ```python # 如果需要安装指定版本的PyTorch !pip install torch==1.9.0 torchvision torchaudio -f https://download.pytorch.org/whl/torch_stable.html ``` #### 导入必要的模块 为了能够顺利运行基于PyTorch的模型，首先应当导入所需的Python包： ```python import numpy as np import pandas as pd from sklearn.model_selection import train_test_split import torch import torch.nn as nn import torch.optim as optim from torch.utils.data import DataLoader, Dataset import torchvision.transforms as transforms from PIL import Image ``` #### 数据集准备与加载 对于图像分类任务来说，通常会涉及到自定义Dataset类以便更好地管理数据流。下面是一个简单的例子展示如何创建一个继承自`torch.utils.data.Dataset`的数据集对象[^4]。 ```python class CustomImageDataset(Dataset): def __init__(self, annotations_file, img_dir, transform=None, target_transform=None): self.img_labels = pd.read_csv(annotations_file) self.img_dir = img_dir self.transform = transform self.target_transform = target_transform def __len__(self): return len(self.img_labels) def __getitem__(self, idx): img_path = os.path.join(self.img_dir, self.img_labels.iloc[idx, 0]) image = Image.open(img_path).convert('RGB') label = self.img_labels.iloc[idx, 1] if self.transform: image = self.transform(image) if self.target_transform: label = self.target_transform(label) return image, label transform = transforms.Compose([ transforms.Resize((28, 28)), transforms.ToTensor(), ]) dataset = CustomImageDataset( annotations_file='data.csv', img_dir='./images/', transform=transform, ) train_loader = DataLoader(dataset, batch_size=64, shuffle=True) test_loader = DataLoader(test_dataset, batch_size=64, shuffle=False) ``` #### 构建并训练模型 接下来就是构建神经网络架构以及编写训练循环逻辑。这里给出一段简化版的代码片段用于说明基本流程[^3]。 ```python device = 'cuda' if torch.cuda.is_available() else 'cpu' model = YourModelClass().to(device) # 将模型移动到GPU上执行计算 criterion = nn.CrossEntropyLoss() optimizer = optim.Adam(model.parameters(), lr=learning_rate) for epoch in range(num_epochs): running_loss = 0.0 for i, (inputs, labels) in enumerate(train_loader): inputs, labels = inputs.to(device), labels.to(device) # 移动输入张量至设备(CPU/GPU) optimizer.zero_grad() outputs = model(inputs) loss = criterion(outputs, labels) loss.backward() optimizer.step() running_loss += loss.item() print(f'[Epoch {epoch + 1}] Loss: {running_loss / len(train_loader)}') print('Finished Training.') ```