Commit 9e5f0e72 authored by dcuai's avatar dcuai
Browse files

Update README.md

parent 3c36c06b
......@@ -23,13 +23,13 @@ BERT的全称为Bidirectional Encoder Representation from Transformers,是一
拉取镜像:
```plaintext
docker pull image.sourcefind.cn:5000/dcu/admin/base/migraphx:4.0.0-centos7.6-dtk23.04.1-py38-latest
docker pull image.sourcefind.cn:5000/dcu/admin/base/migraphx:4.3.0-ubuntu20.04-dtk24.04.1-py3.8
```
创建并启动容器:
```plaintext
docker run --shm-size 16g --network=host --name=bert_migraphx --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v $PWD/bert_migraphx:/home/bert_migraphx -it <Your Image ID> /bin/bash
docker run --shm-size 16g --network=host --name=bert_migraphx -v /opt/hyhal:/opt/hyhal:ro --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v $PWD/bert_migraphx:/home/bert_migraphx -it <Your Image ID> /bin/bash
# 激活dtk
source /opt/dtk/env.sh
......@@ -41,7 +41,7 @@ source /opt/dtk/env.sh
cd ./docker
docker build --no-cache -t bert_migraphx:2.0 .
docker run --shm-size 16g --network=host --name=bert_migraphx --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v $PWD/bert_migraphx:/home/bert_migraphx -it <Your Image ID> /bin/bash
docker run --shm-size 16g --network=host --name=bert_migraphx -v /opt/hyhal:/opt/hyhal:ro --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v $PWD/bert_migraphx:/home/bert_migraphx -it <Your Image ID> /bin/bash
# 激活dtk
source /opt/dtk/env.sh
......@@ -169,4 +169,4 @@ https://developer.hpccube.com/codes/modelzoo/bert_migraphx
## 参考资料
https://github.com/ROCmSoftwarePlatform/AMDMIGraphX/tree/develop/examples/nlp/python_bert_squad
\ No newline at end of file
https://github.com/ROCmSoftwarePlatform/AMDMIGraphX/tree/develop/examples/nlp/python_bert_squad
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