一、安装前准备
1. github上下载ChatGLM2-6B代码
github地址:https://github.com/THUDM/ChatGLM2-6B.git ChatGLM2-6B GitHub地址
拉取代码
2.安装Anaconda
Anaconda 就是可以便捷获取包且对包能够进行管理,同时对环境可以统一管理的发行版本。Anaconda包含了conda、Python在内的超过180个科学包及其依赖项。
1.访问Anaconda官网,下载安装包,此处以windows为例。
官网地址:https://www.anaconda.com/ Anaconda官网
2.下载windows安装包
3.安装Anaconda
选择默认即可,安装路径可自行选择
3.配置Anaconda环境
打开Anaconda,创建ChatGlm环境
1.选择Create创建环境
2.输入环境名称,选择python版本,此处选择3.10.12版本点击创建
3.创建完成后点击绿色图标,选择Open Terminal打开控制台
安装gcc
1.启动ChatGlm需要gcc,在Anaconda仓库中搜索gcc,找到合适版本,此处使用 win-64 v5.3.0版本
conda install -c conda-forge m2w64-gcc
2.成功后得到如下信息
安装PyTorch
1.https://pytorch.org/get-started/locally/ 下载地址
2.我这里是windows,所以选择windows,选择conda
3.在刚刚安装gcc的窗口执行如下命令安装PyTorch,文件较大,所需时间比较长,安装完成后会自动清空控制太,输出done
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
安装chardet
完成上述操作启动后chatglm后仍提示缺少组件,按着提示安装即可
1.仍然在上述窗口下执行如下安装命令,安装完成后结果如下
pip install chardet
下载模型
1.由于机器配置,仅以chatglm2-6b-int4模型为例
Hungging Face 模型下载地址:https://huggingface.co/THUDM/chatglm2-6b-int4/tree/main
2.下载除第一个文件外剩下十个文件,可在克隆的项目目录下新建
THUDM/chatglm2-6b-int4 两级目录,将文件存放于chatglm2-6b-int4目录下,部分大文件可从清华大学云盘下载,速度更快
清华云盘下载地址:https://cloud.tsinghua.edu.cn/d/674208019e314311ab5c/?p=%2Fchatglm2-6b-int4&mode=list
下载完成后如下所示
二、运行ChatGlm
1.修改配置
1.此处web_demo.py为例,其它同理
修改红色部分为自己的模型路径,如显存大于13GB可以不用添加“.quantize(4)”。当然,有足够的硬件可以运行标准模型
2.运行ChatGlm
1.打开Anaconda,选择前面创建好的环境重新进入控制台
2.进入到项目目录
3.运行ChatGlm
python web_demo.py
4.运行成功,系统自动打开浏览器,进入到chat界面
5.进行对话