百度360必应搜狗淘宝本站头条
当前位置:网站首页 > 编程字典 > 正文

一次失败部署实践,阿里云免费GPU部署LLaMA2大模型及对话评测

toyiye 2024-06-21 12:19 15 浏览 0 评论

Meta开放了LLaMA 2模型,并且可以直接商用,同时开放了生成版本和聊天版本,包括7b,13b和70b三种规格的大模型。本文接下来将介绍使用阿里云免费GPU部署LLaMA 2模型及进行Llama-2-70b大模型对话能力评测。

Llama-2系列模型下载:

模型下载,常规方式:

准备模型(7b,13b和70b三种规格的大模型类似)

首先,下载 Hugging Face 格式的权重。下载之前记得去 Hugging Face 申请 LLaMA-2 的权重并生成自己的 Access Token。

如没有账号先注册

填写注册信息

允许共享账号信息给meta, 并且会给meta账号绑定的邮箱发邮件

llama-2-70b也同理

Submit之后会给邮箱发一份确认邮件,注册成功之后就可以登录meta申请accessTokens

有了模型权重的访问权限后,我们就能快乐地下载模型啦。

模型下载,使用阿里云的模型镜像方式:

import os
dsw_region = os.environ.get("dsw_region")
url_link = {
"cn-shanghai": "https://atp-modelzoo-sh.oss-cn-shanghai-internal.aliyuncs.com/release/tutorials/llama2/llama2-7b.tar.gz",
"cn-hangzhou": "https://atp-modelzoo.oss-cn-hangzhou-internal.aliyuncs.com/release/tutorials/llama2/llama2-7b.tar.gz",
"cn-shenzhen": "https://atp-modelzoo-sz.oss-cn-shenzhen-internal.aliyuncs.com/release/tutorials/llama2/llama2-7b.tar.gz",
"cn-beijing": "https://atp-modelzoo-bj.oss-cn-beijing-internal.aliyuncs.com/release/tutorials/llama2/llama2-7b.tar.gz", 
}
path = url_link[dsw_region]
os.environ['LINK_CHAT'] = path
!wget $LINK_CHAT
!tar -zxvf llama2-7b.tar.gz


使用免费的阿里云GPU来部署llama2实践:

创建阿里云资源,V100的GPU,32G内存。

新建notebook文件,

执行以下llama2部署脚本:

import gradio as gr
import requests
import json
from transformers import AutoTokenizer, AutoModelForCausalLM

# 模型地址替换为自己训练好的模型地址
tokenizer = AutoTokenizer.from_pretrained("/mnt/workspace/llama2-7b",trust_remote_code=True) 

# 模型地址替换为自己训练好的模型地址
model = AutoModelForCausalLM.from_pretrained("/mnt/workspace/llama2-7b",trust_remote_code=True).eval().half().cuda()

def inference(text):
    from transformers import pipeline
    pipe = pipeline("text-generation", model=model, tokenizer=tokenizer,device='cuda:0', max_new_tokens=400)
    res=pipe(text)
    return res[0]['generated_text'][len(text):]
    
demo = gr.Blocks()

with demo:
    input_prompt = gr.Textbox(label="请输入需求", value="请以算法工程师的身份,写一篇关于大模型发展的发言稿。", lines=6)
    generated_txt = gr.Textbox(lines=6)
    b1 = gr.Button("发送")
    b1.click(inference, inputs=[input_prompt], outputs=generated_txt)
    
demo.launch(enable_queue=True, share=True)

看右上角资源监控,内存狂奔到98.5%。

然后就失败了,系统内存空间不够,GPU还没开始就结束了。

查看资料,llama2-7b模型需要至少64G内存和24GiB及以上的显存。确实内存不够啊,但免费阿里云GPU资源最多只能申请到32G内存,看来免费方式走不通啊。

如果服务器资源够(64G内存和24G显存GPU),以上部署脚本应该可以跑起来。由于资源问题,本次使用阿里云免费GPU部署llama2-7b实验不能继续了,一次失败的实验,失败经验也是知识,故记录下来。


Llama-2-70b的web交互体验:

虽然免费阿里云GPU资源部署llama2模型实践失败,没能进行llama2模型对话体验有点遗憾,但是找到了另一种免费体验llama2大模型对话的方式。

可以访问poe里面的llama2的web对话。

访问地址:https://poe.com/Llama-2-70b

交互体验演示如下:

以上是在poe上面的llama2-70b大模型对话能力的演示,不支持中文,体验不太好。llama2-70b模型本身能力方面,常识性的对话能力还可以,但是推理能力不太行,有待微调提升。


转载:余生不设限 公众号

如果你对这篇文章感兴趣,而且你想要了解更多关于AI领域的实战技巧,可以关注「AIGC研究社」。在这里,你可以看到最新最热的AIGC领域的干货文章和案例实战教程。

欢迎加入技术交流群d1878810988

跟大家推荐下几个AI工具,免费的:

一品AI助手(免费ChatGPT):http://ai.y-p.cc/chat/

一品AI 文案(免费多场景AI文案):http://ai.y-p.cc/text/create.html

一品AI画师(免费AI绘画):http://ai.y-p.cc/aiart/

一品AI工具集:http://ai.y-p.cc

相关推荐

为何越来越多的编程语言使用JSON(为什么编程)

JSON是JavascriptObjectNotation的缩写,意思是Javascript对象表示法,是一种易于人类阅读和对编程友好的文本数据传递方法,是JavaScript语言规范定义的一个子...

何时在数据库中使用 JSON(数据库用json格式存储)

在本文中,您将了解何时应考虑将JSON数据类型添加到表中以及何时应避免使用它们。每天?分享?最新?软件?开发?,Devops,敏捷?,测试?以及?项目?管理?最新?,最热门?的?文章?,每天?花?...

MySQL 从零开始:05 数据类型(mysql数据类型有哪些,并举例)

前面的讲解中已经接触到了表的创建,表的创建是对字段的声明,比如:上述语句声明了字段的名称、类型、所占空间、默认值和是否可以为空等信息。其中的int、varchar、char和decimal都...

JSON对象花样进阶(json格式对象)

一、引言在现代Web开发中,JSON(JavaScriptObjectNotation)已经成为数据交换的标准格式。无论是从前端向后端发送数据,还是从后端接收数据,JSON都是不可或缺的一部分。...

深入理解 JSON 和 Form-data(json和formdata提交区别)

在讨论现代网络开发与API设计的语境下,理解客户端和服务器间如何有效且可靠地交换数据变得尤为关键。这里,特别值得关注的是两种主流数据格式:...

JSON 语法(json 语法 priority)

JSON语法是JavaScript语法的子集。JSON语法规则JSON语法是JavaScript对象表示法语法的子集。数据在名称/值对中数据由逗号分隔花括号保存对象方括号保存数组JS...

JSON语法详解(json的语法规则)

JSON语法规则JSON语法是JavaScript对象表示法语法的子集。数据在名称/值对中数据由逗号分隔大括号保存对象中括号保存数组注意:json的key是字符串,且必须是双引号,不能是单引号...

MySQL JSON数据类型操作(mysql的json)

概述mysql自5.7.8版本开始,就支持了json结构的数据存储和查询,这表明了mysql也在不断的学习和增加nosql数据库的有点。但mysql毕竟是关系型数据库,在处理json这种非结构化的数据...

JSON的数据模式(json数据格式示例)

像XML模式一样,JSON数据格式也有Schema,这是一个基于JSON格式的规范。JSON模式也以JSON格式编写。它用于验证JSON数据。JSON模式示例以下代码显示了基本的JSON模式。{"...

前端学习——JSON格式详解(后端json格式)

JSON(JavaScriptObjectNotation)是一种轻量级的数据交换格式。易于人阅读和编写。同时也易于机器解析和生成。它基于JavaScriptProgrammingLa...

什么是 JSON:详解 JSON 及其优势(什么叫json)

现在程序员还有谁不知道JSON吗?无论对于前端还是后端,JSON都是一种常见的数据格式。那么JSON到底是什么呢?JSON的定义...

PostgreSQL JSON 类型:处理结构化数据

PostgreSQL提供JSON类型,以存储结构化数据。JSON是一种开放的数据格式,可用于存储各种类型的值。什么是JSON类型?JSON类型表示JSON(JavaScriptO...

JavaScript:JSON、三种包装类(javascript 包)

JOSN:我们希望可以将一个对象在不同的语言中进行传递,以达到通信的目的,最佳方式就是将一个对象转换为字符串的形式JSON(JavaScriptObjectNotation)-JS的对象表示法...

Python数据分析 只要1分钟 教你玩转JSON 全程干货

Json简介:Json,全名JavaScriptObjectNotation,JSON(JavaScriptObjectNotation(记号、标记))是一种轻量级的数据交换格式。它基于J...

比较一下JSON与XML两种数据格式?(json和xml哪个好)

JSON(JavaScriptObjectNotation)和XML(eXtensibleMarkupLanguage)是在日常开发中比较常用的两种数据格式,它们主要的作用就是用来进行数据的传...

取消回复欢迎 发表评论:

请填写验证码