Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
MiniCPM_pytorch
Commits
2aab56dd
Commit
2aab56dd
authored
Mar 01, 2024
by
chenzk
Browse files
v1.0.2
parent
a54c4e73
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
4 deletions
+4
-4
README.md
README.md
+4
-4
No files found.
README.md
View file @
2aab56dd
...
@@ -14,7 +14,7 @@
...
@@ -14,7 +14,7 @@
## 环境配置
## 环境配置
```
```
mv
M
ini
CPM
_pytorch MiniCPM # 去框架名后缀
mv
m
ini
cpm
_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
```
```
## 数据集
## 数据集
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment