探索pytorch数据集中Mnist数据集的数据格式

1 问题

1.1安装pytorch时,由于使用的vscode编译器,所以采用pip进行安装,但是遇到pytorch版本与python版本不对应的问题。

1.2探索pytorch数据集中Mnist数据集的数据格式。

2 方法

|------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
| 2.1 首先查看自己电脑能够适应的pytorch版本 2.2 打开pytorch的官网,选择对应的版本 2.3 最后在vscode的终端运行复制的pip代码。(下载pytorch命令:pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117) 2.4最后检测pytorch是否安装成功,出现这个问题 我就重新安装更高的python版本,最后成功解决问题。 2.5 Mnist数据集的格式探索 import matplotlib.pyplot as plt from torchvision import datasets ds = datasets.MNIST( root='data', # 数据集下载的路径 download=True, ) print(ds[0][0]) plt.imshow(ds[0][0]) plt.show() |
| |

3 结语

3.1针对pytorch环境配置的问题,提出更新python版本的方法,通过引入pytorch自带数据集,调用其数据验证pytorch环境是否安装正确的实验,证明该方法是有效的;

3.2探索Minst数据集时,了解二元组的构成。

3.3本文的方法还是有不足之处:就是pytorch为什么要求高版本python的问题,还有就是除了更新python版本之外,应该还有其他可以搭建好pytorch环境的方法,如不使用vscode编辑器。

相关推荐
balmtv2 分钟前
Gemini 3.1 Pro镜像技术拆解:2026年最强推理模型的国内实测
人工智能
roman_日积跬步-终至千里3 分钟前
【深度学习】BatchNorm详解:原理·四步·梯度推导
人工智能·深度学习
纤纡.6 分钟前
实战 OpenCV:从文档扫描到目标追踪,四大核心场景全解析
人工智能·opencv·计算机视觉
gaozhiyong081310 分钟前
Claude 4.6官网Agent Teams架构深度解析:2026年多智能体协作编程实战指南
人工智能
flying_131411 分钟前
图神经网络分享系列-HAN(Heterogeneous Graph Attention Network)(三)
深度学习·神经网络·tensorflow·gat·han·节点级别注意力·语义级别注意力
roman_日积跬步-终至千里12 分钟前
【大语言模型基础(2)】自注意力与多头机制:QKV、缩放与因果掩码
人工智能·语言模型·自然语言处理
明月照山海-13 分钟前
机器学习周报三十八
人工智能·机器学习
不熬夜的熬润之15 分钟前
KCF算法解析
人工智能·算法·计算机视觉·机器人
Mintopia15 分钟前
AI 生成代码的“债务清单”:哪些地方省下的时间,最后会加倍还
人工智能
Bruce204899816 分钟前
技术小白AI入门指南(从0到1,无缝衔接技术基础)
人工智能