8G 显存玩转书生大模型 Demo第二节

本关是部署1.8模型的部署

准备步骤

创建开发机和安装相关依赖来创建环境，需要用到以下命令

登录后复制

# 创建环境
conda create -n demo python=3.10 -y
# 激活环境
conda activate demo
# 安装 torch
conda install pytorch==2.1.2 torchvision==0.16.2 torchaudio==2.1.2 pytorch-cuda=12.1 -c pytorch -c nvidia -y
# 安装其他依赖
pip install transformers==4.38
pip install sentencepiece==0.1.99
pip install einops==0.8.0
pip install protobuf==5.27.2
pip install accelerate==0.33.0
pip install streamlit==1.37.0

• 1.
• 2.
• 3.
• 4.
• 5.
• 6.
• 7.
• 8.
• 9.
• 10.
• 11.
• 12.
• 13.

安装完之后运行Python代码部署模型

运行结果如下

运用Streamlit Web Demo进行可视化展示

运行下面streamlit run /root/demo/Tutorial/tools/streamlit_demo.py --server.address 127.0.0.1 --server.port 6006
接着通过ssh来映射窗口启动Demo

作业结果

原创作者: u_16507376 转载于: https://blog.51cto.com/u_16507376/11714345