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

相关推荐
番石榴AI12 分钟前
基于机器学习优化的主图选择方法(酒店,景点,餐厅等APP上的主图展示推荐)
图像处理·人工智能·python·机器学习
国产化创客24 分钟前
基于AI大模型智能硬件--小智AI项目PC端部署测试
人工智能
海边夕阳200625 分钟前
【每天一个AI小知识】:什么是零样本学习?
人工智能·经验分享·学习
平凡而伟大(心之所向)29 分钟前
云架构设计与实践:从基础到未来趋势
人工智能·阿里云·系统架构·安全架构
数据与后端架构提升之路30 分钟前
构建一个可进化的自动驾驶数据管道:规则引擎与异常检测的集成
人工智能·机器学习·自动驾驶
qq74223498438 分钟前
Python操作数据库之pyodbc
开发语言·数据库·python
xier_ran1 小时前
Transformer:Decoder 中,Cross-Attention 所用的 K(Key)和 V(Value)矩阵,是如何从 Encoder 得到的
深度学习·矩阵·transformer
2401_841495641 小时前
【自然语言处理】轻量版生成式语言模型GPT
人工智能·python·gpt·深度学习·语言模型·自然语言处理·transformer
梵得儿SHI1 小时前
(第三篇)Spring AI 基础入门:PromptTemplate 与对话工程实战(从字符串拼接到底层模板引擎的进阶之路)
人工智能·prompt·大模型应用·spring ai·prompttemplate·ai 响应的质量与准确性·上下文管理策略