Mistral-7B-v0.3适配昇腾开源验证任务心得

举报
huyanbo 发表于 2024/11/22 09:30:22 2024/11/22
【摘要】 本人验证Mistral-7b-v0.3能不能在 Ascend(昇腾)上运行过程的心得

Mistral-7B-v0.3适配昇腾开源验证任务心得

 

首先看了一下任务计划书:https://bbs.huaweicloud.com/blogs/440092主要目的是将 Mistral-7b-v0.3 项目适配至华为的 Ascend(昇腾)和 Kunpeng(鲲鹏)处理器,确保其在这些硬件平台上的高效运行,也就是说验证Mistral-7b-v0.3能不能在 Ascend(昇腾)上运行。

开发过程

1、下载模型

明确目的后开始熟悉模型,Mistral-7b-v0.3github网址:https://github.com/mistralai/mistral-inference。为了方便后续使用,我把代码库中的代码文件fork到了我自己的代码仓库,代码仓库的README文件有模型的下载链接,也可以去常用的模型仓库网站寻找模型然后下载,比如huggingface和魔塔社区。

我是在魔塔社区下载的模型。先使用:

pip install modelscope

 安装modelscope,然后使用:

modelscope download --model LLM-Research/Mistral-7B-Instruct-v0.3 --local_dir ./dir

把模型下载到当前文件夹。


2、编写推理代码在NPU推理模型

官方并没有直接提供mistral-7b-v0.3的推理代码,需要自己编写,可以参考:https://gitcode.com/huyanbo/mistral-7b-v0.3/overview  

3、安装依赖包

创建python虚拟环境并且安装numpy库(注:numpy版本建议安装1.23左右的,不要安装2.0及以上的,会有兼容问题。)

使用NPU推理模型需要安装pytorchCANNtorch_npu

先安装CANN,参考:https://www.hiascend.com/zh/developer/download/community/result?module=cann

安装pytorch的cpu版本:https://pytorch.org/  pip安装代码:pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu

torch_npu的安装参考:https://github.com/Ascend/pytorch/blob/master/README.zh.md

其他的代码依赖包可以运行pip install -r requirement.txt,把我的代码库里requirement.txt文件中写出的依赖包安装到你的环境之中。其他安装包在运行时按照提示安装即可。

 

1       验收材料


验证DEMO地址:https://gitcode.com/huyanbo/mistral-7b-v0.3/overview


 

2       资源清单

产品名称

NPU架构

CPU架构

操作系统

云堡垒机

Ascend 910B3

 

鲲鹏计算

Kunpeng-920

Huawei Cloud EulerOS 2.0 (aarch64)

 

 

3       测试结果

Npu推理

运行程序前


运行程序后


我编辑npu时指定了npu编号为4,可以看到npu4已经在运行了。

 

推理结果

此模型是对话模型,我输入问题并提交给模型后,模型正确反馈了答案


 

 

【版权声明】本文为华为云社区用户原创内容,转载时必须标注文章的来源(华为云社区)、文章链接、文章作者等基本信息, 否则作者和本社区有权追究责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。