Commit db1de1e2 authored by dcuai's avatar dcuai
Browse files

Update README.md

parent 93a8eaae
...@@ -42,8 +42,9 @@ pip install auto-gptq -i https://pypi.mirrors.ustc.edu.cn/simple/ ...@@ -42,8 +42,9 @@ pip install auto-gptq -i https://pypi.mirrors.ustc.edu.cn/simple/
### Dockerfile(方法二) ### Dockerfile(方法二)
此处提供dockerfile的使用方法 此处提供dockerfile的使用方法
``` ```
cd TeleChat/docker
docker build --no-cache -t telechat_pytorch:latest . docker build --no-cache -t telechat_pytorch:latest .
docker run xxx docker run --shm-size 80g --network=host --name=telechat --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v /opt/hyhal:/opt/hyhal:ro -v <Host Path>:<Container Path> -it telechat_pytorch:latest bash
安装依赖: 安装依赖:
cd TeleChat cd TeleChat
...@@ -53,16 +54,18 @@ pip install optimum -i https://pypi.mirrors.ustc.edu.cn/simple/ ...@@ -53,16 +54,18 @@ pip install optimum -i https://pypi.mirrors.ustc.edu.cn/simple/
pip install auto-gptq -i https://pypi.mirrors.ustc.edu.cn/simple/ pip install auto-gptq -i https://pypi.mirrors.ustc.edu.cn/simple/
``` ```
### Anaconda(方法三) ### Anaconda(方法三)
此处提供本地配置、编译的详细步骤,例如: 1、关于本项目DCU显卡所需的特殊深度学习库可从光合开发者社区下载安装:
- https://developer.hpccube.com/tool/
关于本项目DCU显卡所需的特殊深度学习库可从[光合](https://developer.hpccube.com/tool/)开发者社区下载安装。
``` ```
DTK驱动:dtk24.04 DTK驱动:dtk24.04.1
python:python3.10 python:python3.10
torch:2.1.0
torchvision:0.16.0
``` ```
`Tips:以上dtk驱动、python、paddle等DCU相关工具版本需要严格一一对应`
其它非深度学习库参照requirements.txt安装: `Tips:以上dtk驱动、python、torch等DCU相关工具版本需要严格一一对应。`
2、其它非深度学习库参照requirements.txt安装:
``` ```
安装依赖: 安装依赖:
cd TeleChat cd TeleChat
...@@ -71,6 +74,9 @@ pip install 'ms-swift[llm]' -U -i https://pypi.mirrors.ustc.edu.cn/simple/ ...@@ -71,6 +74,9 @@ pip install 'ms-swift[llm]' -U -i https://pypi.mirrors.ustc.edu.cn/simple/
pip install optimum -i https://pypi.mirrors.ustc.edu.cn/simple/ pip install optimum -i https://pypi.mirrors.ustc.edu.cn/simple/
pip install auto-gptq -i https://pypi.mirrors.ustc.edu.cn/simple/ pip install auto-gptq -i https://pypi.mirrors.ustc.edu.cn/simple/
``` ```
## 数据集
## 模型下载链接 ## 模型下载链接
| **模型版本** | **下载链接** | | **模型版本** | **下载链接** |
......
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