pytorch读取MNIST数据集并显示
#直接下载数据集并读取
1 代码:
import torch
import torchvision
import matplotlib.pyplot as plt #用于显示图片
import os
os.environ["KMP_DUPLICATE_LIB_OK"]="TRUE"
#忽略警告
import warnings
warnings.filterwarnings('ignore')
#选择运行设备
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
#下载数据集