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

v1.0.2

parent a54c4e73
...@@ -14,7 +14,7 @@ ...@@ -14,7 +14,7 @@
## 环境配置 ## 环境配置
``` ```
mv MiniCPM_pytorch MiniCPM # 去框架名后缀 mv minicpm_pytorch MiniCPM # 去框架名后缀
``` ```
### Docker(方法一) ### Docker(方法一)
...@@ -22,8 +22,8 @@ mv MiniCPM_pytorch MiniCPM # 去框架名后缀 ...@@ -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 docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-centos7.6-dtk23.10-py38
# <your IMAGE ID>为以上拉取的docker的镜像ID替换,本镜像为:ffa1f63239fc # <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 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 cd /home/MiniCPM
pip install -r requirements.txt # finetune/requirements.txt pip install -r finetune/requirements.txt # finetune/requirements.txt
# deepspeed、flash_attn2、xformers可从whl.zip文件里获取安装: # 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 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 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 ...@@ -78,7 +78,7 @@ bash patch_xformers.rocm.sh
2、其它非特殊库参照requirements.txt安装 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