Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
UNet_migraphx
Commits
4dfec74c
Commit
4dfec74c
authored
Aug 22, 2024
by
liucong
Browse files
修改readme
parent
829fb728
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
23 additions
and
5 deletions
+23
-5
CMakeLists.txt
CMakeLists.txt
+1
-1
README.md
README.md
+21
-3
docker/Dockerfile
docker/Dockerfile
+1
-1
No files found.
CMakeLists.txt
View file @
4dfec74c
...
...
@@ -17,7 +17,7 @@ set(INCLUDE_PATH ${CMAKE_CURRENT_SOURCE_DIR}/Src/
include_directories
(
${
INCLUDE_PATH
}
)
# 添加依赖库路径
set
(
LIBRARY_PATH
${
CMAKE_CURRENT_SOURCE_DIR
}
/depend/lib
64
/
set
(
LIBRARY_PATH
${
CMAKE_CURRENT_SOURCE_DIR
}
/depend/lib/
$ENV{DTKROOT}/lib/
)
link_directories
(
${
LIBRARY_PATH
}
)
...
...
README.md
View file @
4dfec74c
...
...
@@ -28,13 +28,13 @@ U-Net 的核心原理如下:
拉取镜像:
```
docker pull image.sourcefind.cn:5000/dcu/admin/base/migraphx:4.
0
.0-
centos7.6
-dtk2
3
.04.1-py3
8-latest
docker pull image.sourcefind.cn:5000/dcu/admin/base/migraphx:4.
3
.0-
ubuntu20.04
-dtk2
4
.04.1-py3
.10
```
创建并启动容器:
```
docker run --shm-size 16g --network=host --name=unet_migraphx --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v $PWD/unet_migraphx:/home/unet_migraphx -it <Your Image ID> /bin/bash
docker run --shm-size 16g --network=host --name=unet_migraphx --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v $PWD/unet_migraphx:/home/unet_migraphx
-v /opt/hyhal:/opt/hyhal:ro
-it <Your Image ID> /bin/bash
# 激活dtk
source /opt/dtk/env.sh
...
...
@@ -46,7 +46,7 @@ source /opt/dtk/env.sh
cd ./docker
docker build --no-cache -t unet_migraphx:2.0 .
docker run --shm-size 16g --network=host --name=unet_migraphx --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v $PWD/unet_migraphx:/home/unet_migraphx -it <Your Image ID> /bin/bash
docker run --shm-size 16g --network=host --name=unet_migraphx --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v $PWD/unet_migraphx:/home/unet_migraphx
-v /opt/hyhal:/opt/hyhal:ro
-it <Your Image ID> /bin/bash
# 激活dtk
source /opt/dtk/env.sh
...
...
@@ -86,6 +86,16 @@ python Unet.py
### C++版本推理
注意:当使用操作系统不一样时,CMakeList需要做相应的修改:
```
# ubuntu操作系统
${CMAKE_CURRENT_SOURCE_DIR}/depend/lib64/ 修改为 ${CMAKE_CURRENT_SOURCE_DIR}/depend/lib/
# centos操作系统
${CMAKE_CURRENT_SOURCE_DIR}/depend/lib/ 修改为 ${CMAKE_CURRENT_SOURCE_DIR}/depend/lib64/
```
下面介绍如何运行C++代码示例,C++示例的详细说明见Doc目录下的Tutorial_Cpp.md。
#### 安装Opencv依赖
...
...
@@ -106,6 +116,14 @@ rbuild build -d depend
将依赖库依赖加入环境变量LD_LIBRARY_PATH,在~/.bashrc中添加如下语句:
当操作系统是ubuntu系统时:
```
export LD_LIBRARY_PATH=<path_to_yolov3_migraphx>/depend/lib/:$LD_LIBRARY_PATH
```
当操作系统是centos系统时:
```
export LD_LIBRARY_PATH=<path_to_unet_migraphx>/depend/lib64/:$LD_LIBRARY_PATH
```
...
...
docker/Dockerfile
View file @
4dfec74c
FROM
image.sourcefind.cn:5000/dcu/admin/base/migraphx:4.0.0-centos7.6-dtk23.04.1-py38-latest
\ No newline at end of file
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
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment