Commit 9a6a11a4 authored by liucong's avatar liucong
Browse files

修改readme文档

parent 7c882143
......@@ -30,6 +30,10 @@ 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
# 安装依赖
pip uninstall urllib3
pip install urllib3==1.23 -i https://pypi.tuna.tsinghua.edu.cn/simple
```
### Dockerfile(方法二)
......@@ -63,7 +67,7 @@ cd <path_to_gpt2_migraphx>
cd Python/
# 安装依赖
pip install -r requirements.txt
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
# 运行示例
python gpt2.py
......
FROM sugonhub/migraphx:3.2.1-centos7.6-dtk-23.04.1-py38
RUN source /opt/dtk/env.sh
RUN pip uninstall urllib3
RUN pip install urllib3==1.23 -i https://pypi.tuna.tsinghua.edu.cn/simple
\ No newline at end of file
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