多彩编程 多彩编程MZPH · CODE BLOG
ARTICLE DETAIL

文章详情

深耕前端与后端开发技术的一线实战笔记与踩坑复盘。

ModelScope 本地部署速成:从克隆仓库到三行代码跑通推理

ModelScope 本地部署速成:从克隆仓库到三行代码跑通推理 ModelScope 本地部署速成从克隆仓库到三行代码跑通推理【免费下载链接】modelscopeModelScope: bring the notion of Model-as-a-Service to life.项目地址: https://gitcode.com/GitHub_Trending/mo/modelscope你想做 ModelScope 本地部署结果克隆好仓库、装完核心跑第一行推理代码就报No module named transformers——别慌领域依赖本来就不在核心包里。这篇用 3 张任务卡、10 分钟上下带你把情感分析示例真正跑起来。ModelScope 本地部署前的自检清单动手前先对着下面 5 项过一遍。不用追求高配够用档就能跑通第一次推理舒服档是给之后训练更大模型留的余量。检查项够用舒服怎么查不满足怎么办Python 版本3.103.103.12python --version核心包低于 3.10 装不上先升级 Python内存8GB16GB 以上free -hLinux大模型跑不动就先从 base 小模型入手显卡不需要CPU 能跑4GB 以上显存的 N 卡nvidia-smi没有就 CPU 跑只是慢不阻塞磁盘空间5GB20GB 以上df -h先清理模型缓存删掉可重新下载git任意版本任意版本git --version先装 git拉代码要用显卡这一栏能写不需要是因为本文用到的模型都是 base 规格CPU 完全跑得动N 卡只影响快慢不影响能不能跑。让 ModelScope 本地部署跑起来的 3 张任务卡每张卡都有明确的验收标准不达标就先修它别急着往下走。任务卡一 | 拿到代码建好隔离环境这一步结束时你应当位于项目目录内、虚拟环境已激活where pythonLinux 下which python指向.venv里的解释器。前两条命令拉代码并进入目录后两条在项目里创建名为.venv的虚拟环境并激活它。后面所有安装都只影响这个 venv装坏了删掉目录重来即可不会污染系统 Pythongit clone https://gitcode.com/GitHub_Trending/mo/modelscope cd modelscope python -m venv .venv source .venv/bin/activate⚠️ Windows 没有source命令激活时改用.\.venv\Scripts\activate。 ⚠️ 报python: command not found时Windows 上入口常是py把建环境命令改成py -m venv .venv重试。环境隔离好了下一步的风险就小了一半无论安装成功还是失败都跟系统环境无关。任务卡二 | 装上核心框架这一步结束时终端里import modelscope应当不报错。命令里的.指项目本身pip 会按pyproject.toml的打包配置从本地源码安装 modelscope 包。注意核心只带模型管理相关的小依赖filelock、requests、tqdm 几个torch、transformers 这类大件都不在里面所以这一步会很快pip install . python -c import modelscope; print(OK)⚠️ 安装时提示requires Python 3.10说明 Python 太旧回自检清单升级即可没有别的绕法。 ⚠️ 下载慢得反常基本是网络问题给 pip 配置里加个国内镜像源再重试不是代码问题。看到核心装好了你大概会问那开头的 transformers 报错为什么没出现因为接下来要跑的是 NLP 模型还得把对应领域的包装上——这正是下一张卡的事。任务卡三 | 三行代码跑通第一次推理这一步结束时屏幕应当打印出情感分析结果的字典里面带标签和概率。先装 NLP 领域的依赖。这一步会拉 transformers 这类大件下载量是全程最大的耐心等它跑完pip install .[nlp]然后写三行代码用 StructBERT 的中文情感分析 base 模型from modelscope.pipelines import pipeline clf pipeline(text-classification, modeldamo/nlp_structbert_sentiment-classification_chinese-base) print(clf(这个产品用着很顺手交付也快会推荐给身边的朋友。))首次运行会自动把模型权重下载到本地缓存下载逻辑在 modelscope/hub/file_download.py 里下完之后的推理就完全在本地进行。⚠️ 第一次运行慢是正常的它在下权重不是卡死。 ⚠️ 出现 CUDA 相关报错多半是有 N 卡但 torch 装的是 CPU 版装对应 CUDA 版的 torch或者这个小模型直接用 CPU 也行。 ⚠️ 还报No module named transformers说明这张卡的.[nlp]没装上回头补装。跑通之后路就通了同一个 pipeline 接口对 CV、音频、多模态模型都适用把任务卡二的.[nlp]换成.[cv]或.[audio]再替换代码里的任务名和模型 ID 就行。跑通之后的 3 个加分项⚡提速有 N 卡就装 CUDA 版 torch模型自动走 GPUbase 模型的单句推理能从秒级压到毫秒级。完全离线模型权重下载后都缓存在~/.cache/modelscope下之后推理不再依赖网络敏感数据不出本机。️再往前一步想用自己的数据微调看 modelscope/trainers/ 的 Trainer 接口十多行代码就能起一个小模型的训练。如果你还想深入完整的安装说明与任务教程README_zh.md各领域可运行的示例脚本examples/pytorch/想跳过全部环境配置可以用官方 Docker 镜像Dockerfile 都在 docker/ 目录里想看懂自己的任务逻辑怎么接tests/pipelines 目录下的单测就是能跑通的端到端样例【免费下载链接】modelscopeModelScope: bring the notion of Model-as-a-Service to life.项目地址: https://gitcode.com/GitHub_Trending/mo/modelscope创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考
返回列表