Commit ecb819a2 authored by liucong's avatar liucong
Browse files

修改readme

parent 9a6a11a4
...@@ -26,7 +26,7 @@ GPT-2中使用了掩模自注意力(masked self-attention),通过屏蔽当前 ...@@ -26,7 +26,7 @@ GPT-2中使用了掩模自注意力(masked self-attention),通过屏蔽当前
docker pull sugonhub/migraphx:3.2.1-centos7.6-dtk-23.04.1-py38 docker pull sugonhub/migraphx:3.2.1-centos7.6-dtk-23.04.1-py38
``` ```
创建并启动容器: 创建并启动容器,安装相关依赖
``` ```
docker run --shm-size 16g --network=host --name=gpt2_migraphx --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v $PWD/gpt2_migraphx:/home/gpt2_migraphx -it <Your Image ID> /bin/bash docker run --shm-size 16g --network=host --name=gpt2_migraphx --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v $PWD/gpt2_migraphx:/home/gpt2_migraphx -it <Your Image ID> /bin/bash
...@@ -34,6 +34,9 @@ docker run --shm-size 16g --network=host --name=gpt2_migraphx --privileged --dev ...@@ -34,6 +34,9 @@ docker run --shm-size 16g --network=host --name=gpt2_migraphx --privileged --dev
# 安装依赖 # 安装依赖
pip uninstall urllib3 pip uninstall urllib3
pip install urllib3==1.23 -i https://pypi.tuna.tsinghua.edu.cn/simple pip install urllib3==1.23 -i https://pypi.tuna.tsinghua.edu.cn/simple
# 激活dtk
source /opt/dtk/env.sh
``` ```
### Dockerfile(方法二) ### Dockerfile(方法二)
......
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