Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "d4c9136ce0d6101f52307098595addfcc6a53db5"
Commit 68c40ec3 authored by dcuai's avatar dcuai
Browse files

Update README.md

parent a327b27b
Hide whitespace changes
Inline Side-by-side
Showing with 14 additions and 8 deletions
README.md
View file @ 68c40ec3
...@@ -18,38 +18,41 @@ U-Net 的核心原理如下： ...@@ -18,38 +18,41 @@ U-Net 的核心原理如下：
## 环境配置 ## 环境配置
### Docker（方法一） ### Docker（方法一）
``` ```
docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.13.1-centos7.6-dtk-23.04-py37-latest docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-centos7.6-dtk24.04-py310
docker run -it --network=host --name=unet --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 --ulimit memlock=-1:-1 image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.10.0-centos7.6-dtk-23.04-py37-latest docker run -it --network=host --name=unet --privileged -v /opt/hyhal:/opt/hyhal:ro --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 --ulimit memlock=-1:-1 image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-centos7.6-dtk24.04-py310
cd /path/unet-pytorch cd /path/unet-pytorch
pip install -r requirements.txt pip install -r requirements.txt
pip install NumPy==1.23.5
``` ```
### Dockerfile（方法二） ### Dockerfile（方法二）
``` ```
docker build --no-cache -t unet:latest . docker build --no-cache -t unet:latest .
docker run -dit --network=host --name=unet --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 --ulimit stack=-1:-1 --ulimit memlock=-1:-1 unet:latest docker run -dit --network=host --name=unet --privileged -v /opt/hyhal:/opt/hyhal:ro --device=/dev/kfd --device=/dev/dri --ipc=host --shm-size=16G --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -u root --ulimit stack=-1:-1 --ulimit memlock=-1:-1 unet:latest
docker exec -it unet /bin/bash docker exec -it unet /bin/bash
cd /path/unet-pytorch cd /path/unet-pytorch
pip install -r requirements.txt pip install -r requirements.txt
pip install NumPy==1.23.5
``` ```
### Anaconda（方法三） ### Anaconda（方法三）
关于本项目DCU显卡所需的特殊深度学习库可从[光合](https://developer.hpccube.com/tool/)开发者社区下载安装。 关于本项目DCU显卡所需的特殊深度学习库可从[光合](https://developer.hpccube.com/tool/)开发者社区下载安装。
``` ```
DTK驱动：dtk23.04 DTK驱动：dtk24.04
python：python3.7 python：python3.10
apex：0.1+f49ddd4.abi0.dtk2304.torch1.13 apex：1.1.0+0dd7f68.abi0.dtk2404.torch2.1
torch：1.13.1+git55d300e.abi0.dtk2304 torch：2.1.0+git00661e0.abi0.dtk2404
torchvision：0.14.1+git9134838.abi0.dtk2304.torch1.13 torchvision：0.16.0+gitc9e7141.abi0.dtk2404.torch2.1
``` ```
`Tips：以上dtk驱动、python等DCU相关工具版本需要严格一一对应` `Tips：以上dtk驱动、python等DCU相关工具版本需要严格一一对应`
其它非深度学习库参照requirements.txt安装： 其它非深度学习库参照requirements.txt安装：
``` ```
pip install -r requirements.txt pip install -r requirements.txt
pip install NumPy==1.23.5
``` ```
## 数据集 ## 数据集
...@@ -73,6 +76,9 @@ bash scripts/download_data.sh ...@@ -73,6 +76,9 @@ bash scripts/download_data.sh
``` ```
## 训练 ## 训练
关闭wandb
wandb disabled
### 单机单卡 ### 单机单卡
``` ```
python train.py python train.py
......
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