探索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编辑器。

相关推荐
B站_计算机毕业设计之家3 分钟前
豆瓣电影数据采集分析推荐系统 | Python Vue Flask框架 LSTM Echarts多技术融合开发 毕业设计源码 计算机
vue.js·python·机器学习·flask·echarts·lstm·推荐算法
weixin_446260858 分钟前
掌握 Claude Code Hooks:让 AI 变得更聪明!
人工智能
小白|10 分钟前
CANN性能调优实战:从Profiling到极致优化的完整方案
人工智能
哈__10 分钟前
CANN加速图神经网络GNN推理:消息传递与聚合优化
人工智能·深度学习·神经网络
渣渣苏11 分钟前
Langchain实战快速入门
人工智能·python·langchain
七月稻草人12 分钟前
CANN 生态下 ops-nn:AIGC 模型的神经网络计算基石
人工智能·神经网络·aigc·cann
User_芊芊君子13 分钟前
CANN_MetaDef图定义框架全解析为AI模型构建灵活高效的计算图表示
人工智能·深度学习·神经网络
I'mChloe13 分钟前
CANN GE 深度技术剖析:图优化管线、Stream 调度与离线模型生成机制
人工智能
凯子坚持 c14 分钟前
CANN 生态全景:`cann-toolkit` —— 一站式开发套件如何提升 AI 工程效率
人工智能
lili-felicity16 分钟前
CANN流水线并行推理与资源调度优化
开发语言·人工智能