安装chatglm

假装我不帅

已于 2023-05-01 08:22:26 修改

阅读量1.4k

点赞数

分类专栏： python 文章标签： python 人工智能 linux

于 2023-04-30 17:36:49 首次发布

本文链接：https://blog.csdn.net/qq_36437991/article/details/130448876

版权

python 专栏收录该内容

21 篇文章 0 订阅

订阅专栏

地址

下载源代码

下载完成后解压
在这里插入图片描述

安装cuda

输入nvcc -V查看是否安装cuda
在这里插入图片描述
输入nvidia-smi查看支持的最高版本，最高支持12.1

下载cudahttps://developer.nvidia.com/cuda-downloads

双击安装

同意之后点击下一步

选择精简模式即可

等待下载安装包

输入nvcc -V查看是否安装成功
在这里插入图片描述
其他版本cuda下载
cuda 11.7
cuda 11.8

创建虚拟环境

conda create -p D:\CondaEnvs\chatglmstu  python=3.9
# 激活环境
conda activate D:\CondaEnvs\chatglmstu

安装pytorch

https://pytorch.org/
在这里插入图片描述

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

在这里插入图片描述

下载模型文章

https://huggingface.co/THUDM/chatglm-6b/tree/main
在这里插入图片描述
下载这个网页里的所有文件（一共20个），然后把这些文件放到一个新文件夹，文件夹命名为chatglm-6b，目前默认已经存在这个文件夹

安装其他所需要的软件包

cd D:\pythonstu\ChatGLM-6B-main
pip install -r requirements.txt

在这里插入图片描述

修改项目参数

使用Pycharm打开项目
修改web_demo.py
在这里插入图片描述
THUDM/chatglm-6b修改为chatglm-6b
同理api.py和cli_demo.py也修改
如果你的gpu14G以上不用修改，如果8G选择int8如果6G选择int4

# 4bit
model = AutoModel.from_pretrained("chatglm-6b", trust_remote_code=True).half().quantize(8).cuda()
# 8bit
model = AutoModel.from_pretrained("chatglm-6b", trust_remote_code=True).half().quantize(8).cuda()