双N卡完美运行ChatGLM3

写在前面

之前ChatGLM推出3了,奈何自身条件不允许,CPU运行半天出不来结果,索性放弃。幸好这次拿到了朋友的机器,虽然显存不高,但是双卡共16G也满足运行的最低要求,于是乎准备研究一波,尝试双卡部署一下。

之前已经装好了环境,参考:使用Tesla P4 双卡配置torch机器学习环境

环境配置

关于ChatGLM3不过多介绍了,支持工具回调,支持代码执行,是我挺喜欢的一个开源模型,这次直接给出了技术文档 https://lslfd0slxc.feishu.cn/wiki/HIj5wVxGqiUg3rkbQ1OcVEe5n9g(文档内容不做截图因为有水印)

先下载源码

git clone https://github.com/THUDM/ChatGLM3.git

然后装一下requirements.txt,因为之前已经装好了torch,所以注释掉

pip3 install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple

太慢了,加上国内源

随后,基础环境装好了。

ChatGLM3给了多种运行方式,其中包含基础demo(basic_demo目录下)和综合demo(包含工具执行,代码解释的综合性demo,在composite_demo)

这里我先以方便双卡运行为主,运行basic_demo

关于综合Demo可以参考,本篇不提及:https://github.com/THUDM/ChatGLM3/blob/main/composite_demo/README.md

题外话:运行web_demo.py发现丢了个mdtex2html的库,装一下

  • 21
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

银空飞羽

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值