Skip to content

GitLab

Closed
Created by ached13n1s@ached13n1s

ChatGLM3-6b_pytorch的composite_demo，Code Interpreter无法显示图片

1.docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.13.1-ubuntu20.04-dtk23.10-py310

2.docker run -dit --network=host --name=torch1.13.1-py310 -v /opt/hyhal/:/opt/hyhal/ -v /data:/data --privileged --device=/dev/kfd --device=/dev/dri --ipc=host --shm-size=16G --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -u root --ulimit stack=-1:-1 --ulimit memlock=-1:-1 image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.13.1-ubuntu20.04-dtk23.10-py310

3.docker exec -ti torch1.13.1-py310 /bin/bash

4.git clone http://developer.hpccube.com/codes/modelzoo/chatglm3-6b_pytorch.git

5.cd chatglm3-6b_pytorch

6.pip install -r requirements.txt

  1. cd composite_demo

8..pip install -r requirements.txt

9.ipython kernel install

10.export MODEL_PATH=/data/nfs_ai/AI/05.others/llm/ChatGLM/chatglm3-6b/ #填写真实模型路径

11.export IPYKERNEL=python3

12.streamlit run main.py

进入web界面的Code Interpreter，输入 用python画个爱心

结果不能显示图片