本关是部署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
安装完之后运行Python代码部署模型
运行结果如下
运用Streamlit Web Demo进行可视化展示
运行下面streamlit run /root/demo/Tutorial/tools/streamlit_demo.py --server.address 127.0.0.1 --server.port 6006
接着通过ssh来映射窗口启动Demo
作业结果
