import torchimport torch.nn.functional as Ffrom torch.utils.data import DataLoaderfrom torchvision.datasets import MNISTfrom torchvision import transformsfrom pytorch_lightning import LightningModule, LightningDataModule, Trainerfrom pytor…

・PyTorchPyTorchのCuda対応バージョンを確認。2022/3現在、10.2と11.3。以下では11.3を例に説明。https://pytorch.org/ ・TensorFlow「テスト済みのビルド構成」で上記CUDAバージョンより新しくて、最も近いバージョンを選択。CUDA11.3の例であれば、tensor…