支持pytorch的模型学习环境创建流程(长期更新)

会跟随整个项目进行更新,缺啥找啥就行了,如果这里面也没有说明是非常常见的问题咯。

conda创建

bash 复制代码
conda create -n learn  python=3.10

在终端输入,learn可以改为你要创建的环境名称,python根据自己电脑下载的python进行选择。

torch与cuda载入

你需要先考虑自己应该下什么版本的torch和cuda,去问ai或者查资料。比如我的5070ti就是支持torch2.9.0,cuda12.8。

找好版本后进入网址选好版本https://pytorch.org/get-started/locally/?spm=a2ty_o01.29997173.0.0.2d395171K7OSyy

bash 复制代码
pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu128

cuda版本(GPU),版本可自己通过网站寻找。

bash 复制代码
python -c "import torch; print(torch.__version__); print(torch.cuda.is_available())"

查看下载的pytorch版本和cuda版本确定是否正确。

下载其它pip依赖

其实到这里已经差不多了,接下来运行你的代码,缺什么依赖就pip什么

modelscope

bash 复制代码
ModuleNotFoundError: No module named 'modelscope'

要用魔搭社区就得pip1.10.0的版本

addict

bash 复制代码
ModuleNotFoundError: No module named 'addict'

addict是一个轻量级 Python 库,允许你像访问属性一样操作字典(dict),常用于配置管理、模型参数组织等场景。一些开源项目(如 MMDetection、MMEngine、某些 ModelScope 模型)会依赖它。

pip就能解决

datasets

bash 复制代码
ModuleNotFoundError: No module named 'datasets'

datasets 是 Hugging Face 提供的一个高效、易用的库,用于:

  • 加载和处理公开数据集(如 GLUE、SQuAD、中文 CLUE 等)
  • 创建、保存和共享自定义数据集
  • transformers 无缝集成,常用于模型训练/评估流程

很多 Hugging Face 官方示例、微调脚本都依赖它。

bash 复制代码
pip install datasets==2.21.0

推荐用2.21.0不然和魔塔不兼容

simplejson

bash 复制代码
pip install simplejson

json格式处理的库,学习模型几乎必备。

transformers

python 复制代码
pip install transformers==4.31.0

学习模型必备

相关推荐
GEO AI搜索优化助手5 分钟前
AI搜索革命:营销新纪元,GEO时代生成式AI重构搜索
人工智能·搜索引擎·生成式引擎优化·ai优化·geo搜索优化
赵长辉1 小时前
AGI-rag学习:ChromaDB使用1,txt类型文档【20251016课复习】
python·学习·llm·agi
丝瓜蛋汤1 小时前
NCE(noise contrastive estimation)loss噪声对比估计损失和InfoNCE loss
人工智能
DeepVis Research1 小时前
【AGI Safety/Robotics】2026年度 AGI 对抗性强化学习与软体机器人控制基准索引 (Skynet/Legion Core)
人工智能·网络安全·机器人·数据集·强化学习
赵长辉1 小时前
AGI-rag学习:ChromaDB使用2,json类型文档,问答类内容,【20251016课复习】
学习·agi
Tipriest_7 小时前
torch训练出的模型的组成以及模型训练后的使用和分析办法
人工智能·深度学习·torch·utils
QuiteCoder7 小时前
深度学习的范式演进、架构前沿与通用人工智能之路
人工智能·深度学习
周名彥7 小时前
### 天脑体系V∞·13824D完全体终极架构与全域落地研究报告 (生物计算与隐私计算融合版)
人工智能·神经网络·去中心化·量子计算·agi
MoonBit月兔8 小时前
年终 Meetup:走进腾讯|AI 原生编程与 Code Agent 实战交流会
大数据·开发语言·人工智能·腾讯云·moonbit
小小星球之旅8 小时前
CompletableFuture学习
java·开发语言·学习