安装后或升级启智环境到飞桨2.6版本(develop)

作者 : admin 本文共1106个字,预计阅读时间需要3分钟 发布时间: 2024-06-10 共1人阅读

启智社区的启智大脑调试环境非常好用,但是里面的飞桨环境版本比较低,为了能够运行大模型(llm),需要升级飞桨到2.6版本或者开发版本。

首先创建启智大脑调试环境,注意选cuda12.1的环境。

进入调试环境后,执行:

python -m pip install paddlepaddle-gpu==0.0.0.post120 -f https://www.paddlepaddle.org.cn/whl/linux/gpu/develop.html

安装好后,看看版本:

import paddle
paddle.utils.run_check()
print(paddle.version.full_version)

会显示0.0.0版本,这就是开发板默认的版本显示。

现在准备测试下最新的PaddleNLP是否可以用,先安装PaddleNLP

pip install paddlenlp -i https://mirror.baidu.com/pypi/simple

然后测试一下

# test for paddlenlp
from paddlenlp.transformers import *

tokenizer = AutoTokenizer.from_pretrained("ernie-3.0-tiny-medium-v2-zh")

# 用于分类任务(本项目中的意图识别任务)
seq_cls_model = AutoModelForSequenceClassification.from_pretrained("ernie-3.0-tiny-medium-v2-zh")

# 用于序列标注任务(本项目中的槽位填充任务)
token_cls_model = AutoModelForTokenClassification.from_pretrained("ernie-3.0-tiny-medium-v2-zh")

# 用于阅读理解任务
qa_model = AutoModelForQuestionAnswering.from_pretrained("ernie-3.0-tiny-medium-v2-zh")

只要`from paddlenlp.transformers import *`这句没有报错,那么就证明飞桨最新版本安装成功,且PaddleNLP也能正常使用了。

PaddleNLP的飞桨大模型套件秉承了一站式体验、性能极致、生态兼容的设计理念,旨在提供业界主流大模型预训练、精调(含SFT、PEFT)、量化、推理等统一流程, 帮助开发者低成本、低门槛、快速实现大语言模型定制化。

PaddleNLP大模型套件支持LlaMA、ChatGLM以及Qwen等多种业内流行开源大模型,大家快来试试吧!

安装后或升级启智环境到飞桨2.6版本(develop)插图

本站无任何商业行为
个人在线分享 » 安装后或升级启智环境到飞桨2.6版本(develop)
E-->