修改readme

CMakeLists.txt

@@ -12,15 +12,9 @@ set(CMAKE_BUILD_TYPE release)
 # 添加头文件路径
 set(INCLUDE_PATH    ${CMAKE_CURRENT_SOURCE_DIR}/Src/
                     ${CMAKE_CURRENT_SOURCE_DIR}/Src/Utility/ 
-                    $ENV{DTKROOT}/include/
-                    ${CMAKE_CURRENT_SOURCE_DIR}/depend/include/)
+                    $ENV{DTKROOT}/include/)
 include_directories(${INCLUDE_PATH})
 
-# 添加依赖库路径
-set(LIBRARY_PATH ${CMAKE_CURRENT_SOURCE_DIR}/depend/lib64/
-                $ENV{DTKROOT}/lib/)
-link_directories(${LIBRARY_PATH})
-
 # 添加依赖库
 set(LIBRARY migraphx
             migraphx_gpu

README.md

@@ -23,13 +23,13 @@ GPT-2中使用了掩模自注意力(masked self-attention)，通过屏蔽当前
 拉取镜像：
 
 ```
-docker pull image.sourcefind.cn:5000/dcu/admin/base/migraphx:4.0.0-centos7.6-dtk23.04.1-py38-latest
+docker pull image.sourcefind.cn:5000/dcu/admin/base/migraphx:4.3.0-ubuntu20.04-dtk24.04.1-py3.10
 ```
 
 创建并启动容器，安装相关依赖：
 
 ```
-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 -v /opt/hyhal:/opt/hyhal:ro -it <Your Image ID> /bin/bash
 
 # 激活dtk
 source /opt/dtk/env.sh
@@ -41,7 +41,7 @@ source /opt/dtk/env.sh
 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
+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 -v /opt/hyhal:/opt/hyhal:ro -it <Your Image ID> /bin/bash
 
 # 激活dtk
 source /opt/dtk/env.sh

docker/Dockerfile

-FROM image.sourcefind.cn:5000/dcu/admin/base/migraphx:4.0.0-centos7.6-dtk23.04.1-py38-latest
+FROM image.sourcefind.cn:5000/dcu/admin/base/migraphx:4.3.0-ubuntu20.04-dtk24.04.1-py3.10
\ No newline at end of file