修改镜像名

README.md

@@ -18,12 +18,12 @@ GPT-2中使用了掩模自注意力(masked self-attention)，通过屏蔽当前
 
 ## 环境配置
 
-### Docker（方法一）
+### Docker
 
 拉取镜像：
 
 ```
-docker pull sugonhub/migraphx:3.2.1-centos7.6-dtk-23.04.1-py38
+docker pull image.sourcefind.cn:5000/dcu/admin/base/migraphx:4.0.0-centos7.6-dtk23.04.1-py38-latest
 ```
 
 创建并启动容器，安装相关依赖：
@@ -31,23 +31,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
-
 # 激活dtk
 source /opt/dtk/env.sh
 ```
 
-### Dockerfile（方法二）
-
-```
-cd ./docker
-docker build --no-cache -t gpt2_migraphx:2.0 .
-
-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
-```
-
 ## 数据集
 
 采用交互式界面，通过输入开头诗词，GPT2模型可以推理出后续的诗句。

docker/Dockerfile

-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

model.properties

 # 模型唯一标识
-modelCode = 206
+modelCode=206
 # 模型名称
 modelName=gpt2_migraphx
 # 模型描述