在Windows11的WSL上运行Llama2-7b-chat 下

        上一篇博客讲了我跑Llama的demo的心路历程(上一篇博客传送门),这篇我们主要是讲下怎么配置。

快速开始

        使用Linux、Linux、Linux,重要的事情说三遍,如果你和我一样懒得安装双系统,那么在Windows下安装一个WSL2。我假设大家都有了WSL2,并且开了ubuntu系统,我们需要做如下操作:

        1. 安装Anaconda:如何在Linux服务器上安装Anaconda(超详细) 感谢这位博主

        2. 配置git的代理:WSL2 cannot access Windows proxy、WSL2网络代理配置(apt与git)

        3. 获取代码,获取的方式很多反正能拿到代码就行:git clone https://github.com/facebookresearch/llama.git

        4. 使用代码仓库的  download.sh 下载模型,下载之前先去这个网站填表Meta模型下载申请。开源了个寂寞,扭扭捏捏的。

  •         下载模型请选择:llama-2-7b-chat
  •         太贪了什么都得不到。。。

        5. 安装Llama所需的各种库

                        pip install -e .

        6. 运行demo

torchrun --nproc_per_node 1 example_chat_completion.py \
    --ckpt_dir llama-2-7b-chat/ \
    --tokenizer_path tokenizer.model \
    --max_seq_len 512 --max_batch_size 6

         7. 官方demo没有持续对话,需要自己做修改,按照他们的格式改,我改过的输出如下:

在Windows11的WSL上运行Llama2-7b-chat 下_第1张图片

        8. 我倒是想用gradio做一个页面,但是我没有界面。。。。后续有时间再看

结尾

这就是我在Windows上运行Llama的经历了,血泪教训:用Linux!!!

如果觉得文章对你有用请点赞、关注  ->> 你的点赞对我太有用了
群内交流更多技术
130856474  <--  在这里

你可能感兴趣的:(开源大模型使用,WSL,pytorch,gpt,llama)