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

修改readme文档

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