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

相关推荐
马拉AI3 分钟前
Transformer范式改变?稀疏线性混合SALA架构发布,单卡5090跑通百万长文!
深度学习·架构·transformer
Eric2234 分钟前
CLI-Agent-Manager:面向 Vibe Coding 的多 Agent 统一管理面板
人工智能·后端·开源
如若1238 分钟前
SoftGroup训练FORinstance森林点云数据集——从零到AP=0.506完整复现
人工智能·python·深度学习·神经网络·计算机视觉
InternLM10 分钟前
LMDeploy重磅更新:从支撑模型到被模型反哺,推理引擎迈入协同进化时代!
人工智能·大模型·多模态大模型·大模型推理·书生大模型
AI周红伟10 分钟前
周红伟老师《企业级 RAG+Agent+Skills+OpenClaw 智能体实战内训》的完整课程大纲(5 天 / 40 小时)
人工智能
火红色祥云24 分钟前
深度学习入门:基于Python的理论与实现笔记
笔记·python·深度学习
FoldWinCard24 分钟前
Python 第五次作业
linux·windows·python
hit56实验室25 分钟前
【易经系列】《蒙卦》六五:童蒙,吉。
人工智能
AI浩26 分钟前
VISION KAN:基于Kan的无注意力视觉骨干网络
人工智能·目标检测
China_Yanhy26 分钟前
转型AI运维工程师·Day 10:拥抱“不确定性” —— 断点续训与 Spot 实例抢占
运维·人工智能·python