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

启智社区的启智大脑调试环境非常好用,但是里面的飞桨环境版本比较低,为了能够运行大模型(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等多种业内流行开源大模型,大家快来试试吧!

相关推荐
别烦姐在搞大模型几秒前
一文带你 "看见" MCP 的过程,彻底理解 MCP 的概念
人工智能
用户711283928478 分钟前
LangChain(三) LCEL
人工智能·langchain
愚农搬码20 分钟前
SpringBoot集成Milvus向量库实现CRUD
数据库·人工智能
AI大模型知识23 分钟前
如何微调推理大模型?以Qwen3/DeepSeek-R1为例
人工智能·llm
于齐龙25 分钟前
什么是AI芯片?
人工智能
渝欢意25 分钟前
机器学习——主成分分析(PCA)
人工智能·机器学习
机器之心27 分钟前
重磅开源!首个全异步强化学习训练系统来了,SOTA推理大模型RL训练提速2.77倍
人工智能
leo030828 分钟前
torch.distributed.launch 、 torchrun 和 torch.distributed.run 无法与 nohup 兼容
人工智能·pytorch·python·深度学习·分布式训练
请输入验证码30 分钟前
PyTorch 入门学习笔记(数字识别实战)
人工智能·pytorch·python·神经网络
layneyao30 分钟前
AI数据集构建:从爬虫到标注的全流程指南
人工智能·爬虫