Commit ecb819a2 authored by liucong's avatar liucong
Browse files

修改readme

parent 9a6a11a4
......@@ -26,7 +26,7 @@ GPT-2中使用了掩模自注意力(masked self-attention),通过屏蔽当前
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
......@@ -34,6 +34,9 @@ docker run --shm-size 16g --network=host --name=gpt2_migraphx --privileged --dev
# 安装依赖
pip uninstall urllib3
pip install urllib3==1.23 -i https://pypi.tuna.tsinghua.edu.cn/simple
# 激活dtk
source /opt/dtk/env.sh
```
### 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