AI小白使用Macbook Pro安装llama3与langchain初体验

最新推荐文章于 2024-07-12 10:45:10 发布

AI-椰子不椰

最新推荐文章于 2024-07-12 10:45:10 发布

阅读量1.1k

点赞数 32

文章标签：人工智能 langchain 产品经理大模型 AI大模型

本文链接：https://blog.csdn.net/2401_85343303/article/details/140038851

版权

1. 背景

AI爆火了2年有余，但我仍是一个AI小白，最近零星在学，随手记录点内容供自己复习。

上次在Macbook Pro上安装了Stable Diffusion，体验了本地所心所欲地生成各种心仪的图片，完全没有任何限制的惬意。今天想使用Macbook Pro安装一个本地大语言模型体验一下，刚好在2024年4月18日，Meta在官网上宣布公布了旗下最新大模型Llama 3，并开放了80亿（8b）和700亿（70b）两个小参数版本，据说能力显著提升。遂开干。

为什么部署本地大模型
- 学习方便，私有材料不用发给外网，可以为公司私有化部署积累经验。
- 省钱，不需要单独买云主机，电脑放家里闲着也是闲着。
为什么选择llama3
- 最新款。科技这东西，用新不用旧。
- Meta出品，大厂品质有保证。
这是本次用到的技术框架

2. 环境

硬件：

型号：macbook pro 14寸

CPU：M2 MAX （12+38）

内存：96G

硬盘：8T

操作系统：maxOS 14.3.1

软件：

python 3.11

conda 24.3.0

llama3 8B 和 70B

3. 安装llama3

登录官网：[github.com/ollama/olla…]

下载安装包：[ollama.com/download/Ol…]

解压后运行：Ollama，初始化环境。

先体验8B模型，在命令行窗口运行(第一次运行会下载并安装模型)：

ollama run llama3

安装完成后，输出提示“end a message (/? for help)，可以随便输入信息。

对中文支持还不错。

4. 使用langchain完成简单的RAG

上面对广州的介绍输出非常简单，如果想使用自己语料库来完成，比如公司内部有自己的知识库，需要结合公司的知识库来回答问题，那就可以试试langchain。

简单地说，langchain 是一个帮助在应用程序中使用大型语言模型（LLM）的编程框架，可极大简化对LLM的调用。

详细介绍可参考官方文档：[python.langchain.com/docs/get_st…]

快速开始：[python.langchain.com/docs/get_st…]

安装：

conda install langchain -c conda-forge

报错：“Verifying transaction: / WARNING conda.core.path_actions:verify(1055): Unable to create environments file. Path not writable.”

说明没有写权限，把对应文件owner修改为当前登录用户：

sudo chown -R $USER ~/.conda

为了方便，使用PyCharm来写测试文件。

上面使用conda安装langchain的环境，所以新建项目的环境选择“Select existing”，再选择conda。

写一个python文件测试：

from langchain_community.llms import Ollama

llm = Ollama(model="llama3")
response = llm.invoke("使用中文介绍一下广州")
print(response)

报错，提示要安装llama2，根据提示修改文件“/opt/homebrew/anaconda3/lib/python3.11/site-packages/langchain_community/embeddings/ollama.py”，使用llama3替换：

#model: str = "llama2"
model: str = "llama3"

再次报错，提示要安装faiss，使用命令行安装：

pip install faiss-cpu

终于跑成功，输出如下：

换成百度百科的语料，python文件如下：

from langchain_community.llms import Ollama
from langchain_core.prompts import ChatPromptTemplate
from langchain_community.embeddings import OllamaEmbeddings
from langchain_community.document_loaders import WebBaseLoader
from langchain_community.vectorstores import FAISS
from langchain.text_splitter import RecursiveCharacterTextSplitter
from langchain.chains.combine_documents import create_stuff_documents_chain
from langchain.chains import create_retrieval_chain

#加载文件
loader = WebBaseLoader("https://baike.baidu.com/item/%E5%B9%BF%E5%B7%9E%E5%B8%82/21808?fromtitle=%E5%B9%BF%E5%B7%9E&fromid=72101&fr=aladdin")
page_context = loader.load()
#分词
text_splitter = RecursiveCharacterTextSplitter()
split_documents = text_splitter.split_documents(page_context)

embeddings = OllamaEmbeddings()

#保存到向量库
vector = FAISS.from_documents(split_documents, embeddings)
retriever = vector.as_retriever()

#提示词模板
prompt = ChatPromptTemplate.from_template("""Answer question based on the provided context:
<context>{context}</context>
Question: {input}""")

#加载模型
llm = Ollama(model="llama3")
document_chain = create_stuff_documents_chain(llm, prompt)

retrieval_chain = create_retrieval_chain(retriever, document_chain)
response = retrieval_chain.invoke({"input": "使用中文介绍广州"})
print(response["answer"])

输出如下，可以看出输出的内容部分使用了百度百科最新的数据：

如果把提示词修改一下：“Answer question only based on the provided context”，就是里面加上限定词“only”，输出就只有百度百科的内容，如下：

5. 测试llama3 70b

如果内存足够大，可以选择安装70b模型。70b与8b的区别我现在只知道参数一个多一个少，对硬件要求不同，具体能力区别，还得后面去学习验证。

安装：

ollama run llama3:70b

70b模型安装文件达到了39G多，而8b模型文件是4.7G。

安装完成后，使用命令行测试，输出：

70b模型通过langchain无专属语料输出：

通过对比，在不使用专属语料库的情况下，70b模型比8b模型输出内容更为丰富。

70b模型通过langchain使用专属语料输出：

通过对比，使用专属语料库的情况下，70b模型和8b模型输出内容看不出明显差异。

6. 机器消耗

跑问题时，CPU基本空转，内存跑到64G，GPU打满，风扇呼呼响。

不跑问题时，内存在27G，GPU和CPU负载都很低，风扇不转。

7. 小结

Macbook pro 跑大模型在网上经常被人笑话，不过自己安装玩一玩，学一学，还是不错的，反正我自己用得挺好的。有空的时候再去云平台搞台N卡的机器试试，看到有些云主机平台还有免费试用3个月的带显卡的AI专用虚机供申请，过几天去薅羊毛看看效果。

这也是我第一次写python，果然比java方便。

最后，支付公司都有各种各样的技术文档，可以私有化部署LLM，再结合RAG，私有文档库，做成专有的专家知识库，不仅可以用于外部客户答疑，内部同学值班处理线上问题也是非常方便的。

如何学习AI大模型？

我在一线互联网企业工作十余年里，指导过不少同行后辈。帮助很多人得到了学习和成长。

我意识到有很多经验和知识值得分享给大家，也可以通过我们的能力和经验解答大家在人工智能学习中的很多困惑，所以在工作繁忙的情况下还是坚持各种整理和分享。但苦于知识传播途径有限，很多互联网行业朋友无法获得正确的资料得到学习提升，故此将并将重要的AI大模型资料包括AI大模型入门学习思维导图、精品AI大模型学习书籍手册、视频教程、实战学习等录播视频免费分享出来。

在这里插入图片描述