写点什么

ChatGLM2-6B 环境搭建

作者:IT蜗壳-Tango
  • 2023-07-03
    江苏
  • 本文字数:447 字

    阅读完需:约 1 分钟

本机环境

系统:Win11


显卡:3060 12G


CPU:i5-9400F CPU @ 2.90GHz 2.90 GHz


内存:32G

软件环境

  1. python:3.10.11

  2. git

  3. 魔法上网

项目地址

GitHub - THUDM/ChatGLM2-6B: ChatGLM2-6B: An Open Bilingual Chat LLM | 开源双语对话语言模型

环境搭建

创建虚拟环境


conda create -n ChatGLM2 python=3.10.11 y
复制代码



激活虚拟环境


 conda activate ChatGLM2
复制代码



进入到你想保存项目的位置比如 E 盘的某个文件夹,并克隆项目


git clone https://github.com/THUDM/ChatGLM2-6B.git
复制代码



克隆好项目后,进入到项目目录下


cd ChatGLM2-6B
复制代码



安装 pytorch


conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
复制代码



查看 CUDA 版本


nvidia-smi
复制代码



安装项目依赖文件


pip install -r requirements.txt -i https://mirrors.aliyun.com/pypi/simple/
复制代码


其他常见问题

  1. git clone https://huggingface.co/THUDM/chatglm2-6b报错




解决方案:


git config --global http.proxy 127.0.0.1:10809git config --global https.proxy 127.0.0.1:10809
复制代码


这里设置了 git 的 proxy,其中 10809 是魔法上网的端口号


发布于: 刚刚阅读数: 3
用户头像

一个日语专业的程序猿。 2017-09-10 加入

【坐标】无锡 【软件技能】Java,C#,Python 【爱好】炉石传说 【称号】InfoQ年度人气作者,Intel OpenVINO领航者联盟成员 【B站】https://space.bilibili.com/397260706/ 【个人站】www.it-worker.club

评论

发布
暂无评论
ChatGLM2-6B环境搭建_IT蜗壳-Tango_InfoQ写作社区