PyTorchでMNISTの手書き数字認識モデルを構築する
開発環境設定
WSL2(Ubuntu 22.04) + PyTorch 2.1.2 + Python 3.9.18環境構築
ハードウェアアクセラレーション設定
import torch
# CUDA互換GPUの利用設定
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
print(f"使用デバイス: {device}")
データ処理パイプライン構築
データセット準備
from torchvision import datasets, transforms
# ...
6月20日 16:09 投稿