星海AI-GPU算力云平台部署StoryDiffusion
AI探索家_星海算力 2024-06-24 15:01:18 阅读 59
1.前置准备
StoryDiffusion官方文档中对GPU显存有要求,推荐使用AD103、A4000、Tesla P40等GPU
1.1 创建实例
1.2 从Github获得克隆地址
StoryDiffusion Github地址:https://github.com/HVision-NKU/StoryDiffusion
1.3 启动GPU实例
2.连接服务器并部署StoryDiffusion
2.1 通过终端连接服务器
通过clone获得storyDiffusion,命令如下:
git clone https://github.com/HVision-NKU/StoryDiffusion
通过ls命令查看目录,如果有StoryDiffusion存在,说明克隆成功
使用cd命令进入StoryDiffusion文件夹,运行gradio_app_sdxl_specific_id_low_vram.py文件,根据相应的提示使用pip命令下载缺失的库,命令如下:
cd StoryDiffusion
python gradio_app_sdxl_specific_id_low_vram.py
或者跳到2.2步骤,提前将所有需要的库载入
2.2 下载相对应的库
需要使用pip命令下载transformers、torch、gradio、torchvision、diffusers、accelerate一共六个库
pip install transformerspip install torchpip install gradiopip install torchvisionpip install diffuserspip install accelerate
下载完成后运行gradio_app_sdxl_specific_id_low_vram.py文件,由于使用huggingface时下载连接不稳定,会出现连接失败的提示,因此需要进行一些配置
Huggingface连接配置相关博客地址: https://blog.csdn.net/weixin_44257107/article/details/136532423
2.3 huggingface连接不稳定解决方案
使用nano gradio_app_sdxl_specific_id_low_vram.py命令打开文件进行编辑
nano gradio_app_sdxl_specific_id_low_vram.py
在代码中加入os.environ["HF_ENDPOINT"] ="https://hf-mirror.com/"
os.environ["HF_ENDPOINT"] = "https://hf-mirror.com"
完成修改后,Ctrl+x完成保存,选择Y确定保存,Enter确认修改后的文件名称并回到终端
在终端中设置环境变量HF_ENDPOINT=https://hf-mirror.com python gradio_app_sdxl_specific_id_low_vram.py
HF_ENDPOINT=https://hf-mirror.com python gradio_app_sdxl_specific_id_low_vram.py
如果前面已经添加了accelerate库则不会报错
2.4 运行程序
输入python gradio_app_sdxl_specific_id_low_vram.py运行,可以使用学术加速加快加载速度
python gradio_app_sdxl_specific_id_low_vram.py
学术加速命令 :source /etc/network_turbo
source /etc/network_turbo
如图所示,7860端口成功与本地8888端口建立连接,说明运行成功,本地可通过8888端口访问
3.本地部署WEB访问
3.1 使用终端通过SSH进行访问
3.1.1 打开你的终端或命令提示符
3.1.2 执行SSH命令
3.1.3 假设
远程服务器的IP地址是your-server-ip
。
你在远程服务器上的用户名是root
。
远程服务器上Stable Diffusion应用运行的端口是7
860
。
你想要将此服务映射到本地机器的8888
端口。
3.1.4 命令格式如下
ssh -CNg -L 8888:localhost:7860 root@your-server-ip -p 22
在星海AI-GPU平台创建实例之后,就能在实例界面看见你所创建的实例的ip相关数据
在浏览器地址栏中输入localhost:8888 进行访问,成功加载后的图如下
4.星海算力云
https://ai.spacehpc.com/
欢迎使用星海算力云,星海算力云由北京三轴空间科技有限公司开发,由非盈利组织龙游星海算力产业中心运营的高性能GPU算力云平台。
星海团队长期致力于为图像渲染、科研高性能计算等提供服务。星海AI算力服务平台,获超高速增长,团队规模有100余人,服务了国内AI行业的许多一线团队。
声明
本文内容仅代表作者观点,或转载于其他网站,本站不以此文作为商业用途
如有涉及侵权,请联系本站进行删除
转载本站原创文章,请注明来源及作者。