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
FLAVR_pytorch
Commits
5d2093cb
Commit
5d2093cb
authored
Dec 17, 2024
by
shantf
Browse files
Update the README using the das1.3 image
parent
f130b79e
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
14 additions
and
7 deletions
+14
-7
Dockerfile
Dockerfile
+1
-1
README.md
README.md
+13
-6
hipnn-lib-1128.zip
hipnn-lib-1128.zip
+0
-0
No files found.
Dockerfile
View file @
5d2093cb
FROM
image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-
ubuntu20.04-dtk24.04.1-py3.10
FROM
image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-
py3.10-dtk24.04.3-ubuntu20.04
COPY
requirements.txt requirements.txt
RUN
source
/opt/dtk/env.sh
RUN
cp
/usr/share/zoneinfo/Asia/Shanghai /etc/localtime
&&
echo
'Asia/Shanghai'
>
/etc/timezone
...
...
README.md
View file @
5d2093cb
...
...
@@ -22,8 +22,10 @@
此处提供
[
光源
](
https://sourcefind.cn/#/main-page
)
拉取docker镜像
```
docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-ubuntu20.04-dtk24.04.1-py3.10
docker run -it --network=host --name=flavr --privileged --device=/dev/kfd --device=/dev/dri --ipc=host --shm-size=32G --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -u root --ulimit stack=-1:-1 -v /opt/hyhal:/opt/hyhal:ro --ulimit memlock=-1:-1 image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-ubuntu20.04-dtk24.04.1-py3.10
docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-py3.10-dtk24.04.3-ubuntu20.04
docker run -it --network=host --name=flavr --privileged --device=/dev/kfd --device=/dev/dri --ipc=host --shm-size=32G --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -u root --ulimit stack=-1:-1 -v /opt/hyhal:/opt/hyhal:ro -v path-to-project:/mnt --ulimit memlock=-1:-1 image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-py3.10-dtk24.04.3-ubuntu20.04
path-to-project:项目文件路径;
cd /mnt
pip install -r requirements.txt
```
...
...
@@ -33,7 +35,7 @@ dockerfile使用方法
```
docker build --no-cache -t flavr:latest .
docker run -dit --network=host --name=flavr --privileged --device=/dev/kfd --device=/dev/dri --ipc=host --shm-size=16G --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -u root -v /opt/hyhal:/opt/hyhal:ro --ulimit stack=-1:-1 --ulimit memlock=-1:-1 flavr:latest
docker run -dit --network=host --name=flavr --privileged --device=/dev/kfd --device=/dev/dri --ipc=host --shm-size=16G --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -u root -v /opt/hyhal:/opt/hyhal:ro
-v path-to-project:/mnt
--ulimit stack=-1:-1 --ulimit memlock=-1:-1 flavr:latest
docker exec -it flavr /bin/bash
```
...
...
@@ -42,10 +44,10 @@ docker exec -it flavr /bin/bash
关于本项目DCU显卡所需的特殊深度学习库可从
[
光合
](
https://developer.hpccube.com/tool/
)
开发者社区下载安装。
```
DTK驱动:dtk24.04.
1
DTK驱动:dtk24.04.
3
python:python3.10
torch==2.1.0+das
1.1.git3ac1bdd.abi1
.dtk2404
torchvision==0.16.0+das
1.1.git7d45932.abi1.dtk2404.torch2.1
torch==2.1.0+das
.opt2
.dtk2404
3
torchvision==0.16.0+das
.opt1.dtk24042
```
`Tips:以上dtk驱动、python等DCU相关工具版本需要严格一一对应`
...
...
@@ -56,6 +58,11 @@ torchvision==0.16.0+das1.1.git7d45932.abi1.dtk2404.torch2.1
pip install -r requirements.txt
pip install tensorboard setuptools==57.5.0 six
```
按照下面操作,可以避免因为不使用HIP_VISIBLE_DEVICES=xx,导致报错。
```
unzip /mnt/hipnn-lib-1128.zip
cp /mnt/hipnn-lib-1128/lib/release/* /opt/dtk/lib/
```
## 数据集
...
...
hipnn-lib-1128.zip
0 → 100644
View file @
5d2093cb
File added
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