Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "8f4594074be2b2f8c868f311f24a030806a3deb5"
Commit c05632b7 authored by dcuai's avatar dcuai
Browse files

Update README.md

parent d97792f6
Show whitespace changes
Inline Side-by-side
Showing with 11 additions and 9 deletions
README.md
View file @ c05632b7
...@@ -38,37 +38,39 @@ BERT模型是基于Transformer模型的，但是与原始的Transformer模型不 ...@@ -38,37 +38,39 @@ BERT模型是基于Transformer模型的，但是与原始的Transformer模型不
``` ```
在光源可拉取docker镜像： 在光源可拉取docker镜像：
docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.13.1-centos7.6-dtk-23.04-py37-latest docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-ubuntu20.04-dtk24.04.1-py3.10
创建并启动容器： 创建并启动容器：
docker run -dit --network=host --name=bert4torch --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-py37-latest docker run -dit --network=host --name=bert4torch --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 -v /opt/hyhal:/opt/hyhal:ro image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-ubuntu20.04-dtk24.04.1-py3.10
docker exec -it bert4torch /bin/bash docker exec -it bert4torch /bin/bash
安装依赖包和bert4torch： 安装依赖包和bert4torch：
pip install -r requirements.txt pip install -r requirements.txt
cd bert4torch
python3 setup.py install python3 setup.py install
``` ```
### Dockerfile（方法二） ### Dockerfile（方法二）
``` ```
cd bert4torch/docker
docker build --no-cache -t bert4torch:latest . docker build --no-cache -t bert4torch:latest .
docker run --rm --shm-size 16g --network=host --name=bert4torch --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v $PWD/../../bert4torch:/home/bert4torch -it bert4torch:latest bash docker run --shm-size 16g --network=host --name=bert4torch --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v $PWD/../../bert4torch:/home/bert4torch -v /opt/hyhal:/opt/hyhal:ro -it bert4torch:latest bash
安装依赖包和bert4torch：
pip install -r requirements.txt
python3 setup.py install
``` ```
### Anaconda（方法三） ### Anaconda（方法三）
``` ```
1.创建conda虚拟环境： 1.创建conda虚拟环境：
conda create -n bert4torch python=3.7 conda create -n bert4torch python=3.10
2.关于本项目DCU显卡所需的工具包、深度学习库等均可从光合开发者社区下载安装：https://developer.hpccube.com/tool/ 2.关于本项目DCU显卡所需的工具包、深度学习库等均可从光合开发者社区下载安装：https://developer.hpccube.com/tool/
DTK驱动：dtk23.04 DTK驱动：dtk24.04.1
python：python3.7 python：python3.10
torch:1.13.1 torch:2.1.0
Tips：以上DTK、python、torch等DCU相关工具包，版本需要严格一一对应 Tips：以上DTK、python、torch等DCU相关工具包，版本需要严格一一对应
3.安装bert4torch 3.安装bert4torch
cd bert4torch cd bert4torch
python3 setup.py install python3 setup.py install
4.其它非特殊库参照requirements.txt安装 4.其它非特殊库参照requirements.txt安装
pip install -r 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