Skip to content

GitLab

Commit 0b92dfbe authored by zhaoying1's avatar zhaoying1
Browse files

Update README.md

parent 1fb16033
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
README.md
View file @ 0b92dfbe
...@@ -31,6 +31,8 @@ Baichuan整体模型基于标准的Transformer结构，采用了和LLaMA一样 ...@@ -31,6 +31,8 @@ Baichuan整体模型基于标准的Transformer结构，采用了和LLaMA一样
推荐使用docker方式运行，提供拉取的docker镜像： 推荐使用docker方式运行，提供拉取的docker镜像：
``` ```
docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.13.1-centos7.6-dtk-23.04-py38-latest docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.13.1-centos7.6-dtk-23.04-py38-latest
docker run -dit --network=host --name=baichuan2 --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-centos7.6-dtk-23.04-py38-latest
docker exec -it baichuan2 /bin/bash
``` ```
安装docker中没有的依赖: 安装docker中没有的依赖:
``` ```
...@@ -201,4 +203,4 @@ python cli_demo.py ...@@ -201,4 +203,4 @@ python cli_demo.py
## 参考 ## 参考
- [https://github.com/baichuan-inc/Baichuan2/tree/main](https://github.com/baichuan-inc/Baichuan2/tree/main) - [https://github.com/baichuan-inc/Baichuan2/tree/main](https://github.com/baichuan-inc/Baichuan2/tree/main)
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment