Commit 5d2093cb authored by shantf's avatar shantf
Browse files

Update the README using the das1.3 image

parent f130b79e
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 COPY requirements.txt requirements.txt
RUN source /opt/dtk/env.sh RUN source /opt/dtk/env.sh
RUN cp /usr/share/zoneinfo/Asia/Shanghai /etc/localtime && echo 'Asia/Shanghai' >/etc/timezone RUN cp /usr/share/zoneinfo/Asia/Shanghai /etc/localtime && echo 'Asia/Shanghai' >/etc/timezone
......
...@@ -22,8 +22,10 @@ ...@@ -22,8 +22,10 @@
此处提供[光源](https://sourcefind.cn/#/main-page)拉取docker镜像 此处提供[光源](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 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 --ulimit memlock=-1:-1 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 -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 pip install -r requirements.txt
``` ```
...@@ -33,7 +35,7 @@ dockerfile使用方法 ...@@ -33,7 +35,7 @@ dockerfile使用方法
``` ```
docker build --no-cache -t flavr:latest . 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 docker exec -it flavr /bin/bash
``` ```
...@@ -42,10 +44,10 @@ docker exec -it flavr /bin/bash ...@@ -42,10 +44,10 @@ docker exec -it flavr /bin/bash
关于本项目DCU显卡所需的特殊深度学习库可从[光合](https://developer.hpccube.com/tool/)开发者社区下载安装。 关于本项目DCU显卡所需的特殊深度学习库可从[光合](https://developer.hpccube.com/tool/)开发者社区下载安装。
``` ```
DTK驱动:dtk24.04.1 DTK驱动:dtk24.04.3
python:python3.10 python:python3.10
torch==2.1.0+das1.1.git3ac1bdd.abi1.dtk2404 torch==2.1.0+das.opt2.dtk24043
torchvision==0.16.0+das1.1.git7d45932.abi1.dtk2404.torch2.1 torchvision==0.16.0+das.opt1.dtk24042
``` ```
`Tips:以上dtk驱动、python等DCU相关工具版本需要严格一一对应` `Tips:以上dtk驱动、python等DCU相关工具版本需要严格一一对应`
...@@ -56,6 +58,11 @@ torchvision==0.16.0+das1.1.git7d45932.abi1.dtk2404.torch2.1 ...@@ -56,6 +58,11 @@ torchvision==0.16.0+das1.1.git7d45932.abi1.dtk2404.torch2.1
pip install -r requirements.txt pip install -r requirements.txt
pip install tensorboard setuptools==57.5.0 six 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/
```
## 数据集 ## 数据集
......
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