正文内容 评论(0

6G显存玩转130亿参数大模型 RTX 2060用户发来贺电 AMD不配吗?
2023-05-19 17:24:26  出处:量子位  作者:克雷西 编辑:宪瑞     评论(0)点击可以复制本篇文章的标题和链接

Meta的大语言模型LLaMA 13B,现在用2060就能跑了~

羊驼家族的Alpaca和Vicuna也都能运行,显存最低只需要6G,简直是低VRAM用户的福音有木有。

GitHub上的搭建教程火了之后,网友们纷纷跑来问苹果M2是不是也能跑。

6G显存玩转130亿参数大模型 仅需13行命令 RTX 2060用户发来贺电

这通操作的大致原理是利用最新版CUDA,可以将Transformer中任意数量的层放在GPU上运行。

与此前llama.cpp项目完全运行在CPU相比,用GPU替代一半的CPU可以将效率提高将近2倍。

而如果纯用GPU,这一数字将变成6倍。

6G显存玩转130亿参数大模型 仅需13行命令 RTX 2060用户发来贺电

网友实测的结果中,使用CPU每秒能跑2.1个token,而用GPU能跑3.2个。

6G显存玩转130亿参数大模型 仅需13行命令 RTX 2060用户发来贺电

生成的内容上,开发者成功用它跑出了“尼采文学”。

6G显存玩转130亿参数大模型 仅需13行命令 RTX 2060用户发来贺电

如何操作在开始搭建之前,我们需要先申请获得LLaMA的访问权限。

传送门:https://ai.facebook.com/blog/large-language-model-llama-meta-ai/

此外还需要有一个Linux环境。(Windows用户可以用WSL2)

准备工作完成之后,第一步是将llama.cpp克隆到本地。

6G显存玩转130亿参数大模型 仅需13行命令 RTX 2060用户发来贺电

如果没有安装CUDA,可以参考下面的步骤:

6G显存玩转130亿参数大模型 仅需13行命令 RTX 2060用户发来贺电

然后是建立micromamba环境,安装Python和PyTorch等工具。

接着需要在micromamba环境下安装一些包:

6G显存玩转130亿参数大模型 仅需13行命令 RTX 2060用户发来贺电

然后运行Python脚本以执行转换过程:

6G显存玩转130亿参数大模型 仅需13行命令 RTX 2060用户发来贺电

之后将其量化为4bit模式。

6G显存玩转130亿参数大模型 仅需13行命令 RTX 2060用户发来贺电

接着是新建一个txt文本文档,把提示词输入进去,然后就可以运行了。

6G显存玩转130亿参数大模型 仅需13行命令 RTX 2060用户发来贺电

这步当中-ngl后面的数字是可以修改的,它代表了转换层的数量。

当该值为18时,运行中消耗的VRAM为5.5GB,根据显存的大小最高可以调至40。

网友:AMD不配吗

这一教程出现之后,网友们的新玩具又增加了。

“苦OpenAI久矣”的网友更是感觉仿佛找到了光。

这位网友就表示自己太期待在自己的设备上运行LLM了,宁愿花5千美元购置设备也不想给OpenAI交一分钱。

6G显存玩转130亿参数大模型 仅需13行命令 RTX 2060用户发来贺电

但AMD用户可能就不那么兴奋了,甚至透露出了嫉妒之情。

这套方法要用到CUDA(英伟达专用),所以AMD是不配了吗?

6G显存玩转130亿参数大模型 仅需13行命令 RTX 2060用户发来贺电

那么,你期待用自己的设备跑大语言模型吗?

责任编辑:宪瑞

  • 支持打赏
  • 支持0

  • 反对

  • 打赏

文章价值打分

当前文章打分0 分,共有0人打分
  • 分享好友:
  • |
本文收录在
#NVIDIA#显卡#RTX 2060

  • 热门文章
  • 换一波

  • 好物推荐
  • 换一波

  • 关注我们

  • 微博

    微博:快科技官方

    快科技官方微博
  • 今日头条

    今日头条:快科技

    带来硬件软件、手机数码最快资讯!
  • 抖音

    抖音:kkjcn

    科技快讯、手机开箱、产品体验、应用推荐...