Commit 3d102d61 authored by yangql's avatar yangql
Browse files

新增方法二

parent 9ef4576d
...@@ -18,7 +18,7 @@ Segment Embedding:表明这个词属于哪个句子(NSP需要两个句子) ...@@ -18,7 +18,7 @@ Segment Embedding:表明这个词属于哪个句子(NSP需要两个句子)
## 环境配置 ## 环境配置
### Docker ### Docker(方法一)
拉取镜像: 拉取镜像:
...@@ -34,6 +34,14 @@ docker run --shm-size 16g --network=host --name=bert_ort --privileged --device=/ ...@@ -34,6 +34,14 @@ docker run --shm-size 16g --network=host --name=bert_ort --privileged --device=/
# 激活dtk # 激活dtk
source /opt/dtk/env.sh source /opt/dtk/env.sh
``` ```
### Dockerfile(方法二)
```
cd ./docker
docker build --no-cache -t bert_onnxruntime:2.0 .
docker run --shm-size 16g --network=host --name=bert_ort --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v $PWD/bert_onnxruntime:/home/bert_onnxruntime -it <Your Image ID> /bin/bash
```
## 数据集 ## 数据集
通过对模型提出问题,模型完成问题回答任务。 通过对模型提出问题,模型完成问题回答任务。
......
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