Update README.md

README.md

@@ -18,38 +18,41 @@ U-Net 的核心原理如下：
 ## 环境配置
 ### 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
 pip install -r requirements.txt
+pip install NumPy==1.23.5
 ```
 
 ### Dockerfile（方法二）
 ```
 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
 
 cd /path/unet-pytorch
 pip install -r requirements.txt
+pip install NumPy==1.23.5
 ```
 
 ### Anaconda（方法三）
 关于本项目DCU显卡所需的特殊深度学习库可从[光合](https://developer.hpccube.com/tool/)开发者社区下载安装。
 ```
-DTK驱动：dtk23.04
-python：python3.7
-apex：0.1+f49ddd4.abi0.dtk2304.torch1.13
-torch：1.13.1+git55d300e.abi0.dtk2304
-torchvision：0.14.1+git9134838.abi0.dtk2304.torch1.13
+DTK驱动：dtk24.04
+python：python3.10
+apex：1.1.0+0dd7f68.abi0.dtk2404.torch2.1
+torch：2.1.0+git00661e0.abi0.dtk2404
+torchvision：0.16.0+gitc9e7141.abi0.dtk2404.torch2.1
 ```
 `Tips：以上dtk驱动、python等DCU相关工具版本需要严格一一对应`
 
 其它非深度学习库参照requirements.txt安装：
 ```
 pip install -r requirements.txt
+pip install NumPy==1.23.5
 ```
 
 ## 数据集
@@ -73,6 +76,9 @@ bash scripts/download_data.sh
 ```
 
 ## 训练
+
+关闭wandb
+wandb disabled
 ### 单机单卡
 ```
 python train.py