Quickstart — PyTorch Tutorials 2.6.0+cu124 documentation

安裝torchvision庫

torchvision是PyTorch框架的擴充庫，用於處理影像數據，包括常見的影像變換、資料集載入等功能。 首先，確保你已經安裝PyTorch。 torchvision庫通常與PyTorch一起安裝，但有時可能因為某些原因沒有自動安裝。你可以使用pip或conda來手動安裝它。

使用pip安裝：打開命令列或終端，輸入以下命令來安裝torchvision

pip install torchvision

此外，也出現 ImportError: cannot import name 'ToTensor' 的錯誤。解決方法如下：

我的代碼中出錯的語句是：from albumentations.pytorch import ToTensor

首先，應將其更改為：

from albumentations.pytorch.transforms import ToTensor

接著，在運行時出現新的問題。我根據提示升級 albumentations，具體操作如下：

pip install albumentations==0.5.2

最終，問題得以解決。

使用數據

PyTorch 提供兩種處理數據的基本工具：

torch.utils.data.DataLoader 和 torch.utils.data.Dataset。Dataset 用於存儲樣本及其對應的標籤，而 DataLoader 則將 Dataset 包裝成可迭代的對象。

import torch
from torch import nn
from torch.utils.data import DataLoader
from torchvision import datasets
from torchvision import ToTensor

PyTorch 提供多種領域專用的庫，例如 TorchText、TorchVision 和 TorchAudio，這些庫都內建數據集。在本教程中，我們將使用 TorchVision 提供的數據集。

torchvision.datasets 模組包含許多適用於真實場景的視覺數據集，如 CIFAR、COCO（完整列表請見此處）。在本教程中，我們將使用 FashionMNIST 數據集。每個 TorchVision Dataset 物件都包含兩個參數：transform 和 target_transform，分別用於修改樣本和標籤。