Commit 2aab56dd authored by chenzk's avatar chenzk
Browse files

v1.0.2

parent a54c4e73
......@@ -14,7 +14,7 @@
## 环境配置
```
mv MiniCPM_pytorch MiniCPM # 去框架名后缀
mv minicpm_pytorch MiniCPM # 去框架名后缀
```
### Docker(方法一)
......@@ -22,8 +22,8 @@ mv MiniCPM_pytorch MiniCPM # 去框架名后缀
docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-centos7.6-dtk23.10-py38
# <your IMAGE ID>为以上拉取的docker的镜像ID替换,本镜像为:ffa1f63239fc
docker run -it --shm-size=32G -v $PWD/MiniCPM:/home/MiniCPM -v /opt/hyhal:/opt/hyhal --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name minicpm <your IMAGE ID> bash
cd MiniCPM
pip install -r requirements.txt # finetune/requirements.txt
cd /home/MiniCPM
pip install -r finetune/requirements.txt # finetune/requirements.txt
# deepspeed、flash_attn2、xformers可从whl.zip文件里获取安装:
pip install deepspeed-0.12.3+git299681e.abi0.dtk2310.torch2.1.0a0-cp38-cp38-linux_x86_64.whl
pip install flash_attn-2.0.4_torch2.1_dtk2310-cp38-cp38-linux_x86_64.whl
......@@ -78,7 +78,7 @@ bash patch_xformers.rocm.sh
2、其它非特殊库参照requirements.txt安装
```
pip install -r requirements.txt # finetune/requirements.txt
pip install -r finetune/requirements.txt # finetune/requirements.txt
```
## 数据集
......
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