update

README.md

@@ -22,10 +22,14 @@ mv yolo11_pytorch YOLO11 # 去框架名后缀
 
 ### Docker（方法一）
 ```
-docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.3.0-ubuntu22.04-dtk24.04.2-py3.10
-# <your IMAGE ID>为以上拉取的docker的镜像ID替换，本镜像为：83714c19d308
+docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.5.1-ubuntu22.04-dtk25.04.2-py3.10
+# <your IMAGE ID>为以上拉取的docker的镜像ID替换
 docker run -it --shm-size=64G -v $PWD/YOLO11:/home/YOLO11 -v /opt/hyhal:/opt/hyhal:ro --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name yolo11 <your IMAGE ID> bash
 cd /home/YOLO11
+pip uninstall numpy
+pip uninstall ray
+pip install "numpy<2" -i https://pypi.tuna.tsinghua.edu.cn/simple
+pip install ray==2.37.0
 pip install -r requirements.txt # requirements.txt
 ```
 ### Dockerfile（方法二）
@@ -33,22 +37,21 @@ pip install -r requirements.txt # requirements.txt
 cd cd /home/YOLO11/docker
 docker build --no-cache -t yolo11:latest .
 docker run --shm-size=64G --name yolo11 -v /opt/hyhal:/opt/hyhal:ro --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video -v $PWD/../../YOLO11:/home/YOLO11 -it yolo11 bash
+pip uninstall numpy
+pip uninstall ray
+pip install "numpy<2" -i https://pypi.tuna.tsinghua.edu.cn/simple
+pip install ray==2.37.0
 # 若遇到Dockerfile启动的方式安装环境需要长时间等待，可注释掉里面的pip安装，启动容器后再安装python库：pip install -r requirements.txt。
 ```
 ### Anaconda（方法三）
 1、关于本项目DCU显卡所需的特殊深度学习库可从光合开发者社区下载安装：
 - https://developer.sourcefind.cn/tool/
 ```
-DTK驱动:dtk24.04.2
+DTK驱动:dtk25.04.2
 python:python3.10
-torch:2.3.0
-torchvision:0.18.1
-torchaudio:2.1.2
-triton:2.1.0
-flash-attn:2.0.4
-deepspeed:0.14.2
-apex:1.3.0
-xformers:0.0.25
+torch:2.5.1
+torchvision:0.20.1
+torchaudio:2.5.1
 ```
 
 `Tips：以上dtk驱动、python、torch等DCU相关工具版本需要严格一一对应。`
@@ -56,6 +59,10 @@ xformers:0.0.25
 2、其它非特殊库参照requirements.txt安装
 ```
 cd /home/YOLO11
+pip uninstall numpy
+pip uninstall ray
+pip install "numpy<2" -i https://pypi.tuna.tsinghua.edu.cn/simple
+pip install ray==2.37.0
 pip install -r requirements.txt # requirements.txt
 ```
 

docker/Dockerfile

-FROM image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.3.0-ubuntu22.04-dtk24.04.2-py3.10
+FROM image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.5.1-ubuntu22.04-dtk25.04.2-py3.10
 ENV DEBIAN_FRONTEND=noninteractive
 # RUN yum update && yum install -y git cmake wget build-essential
 # RUN source /opt/dtk-24.04.2/env.sh
 # # 安装pip相关依赖
 COPY requirements.txt requirements.txt
+RUN pip uninstall numpy
+RUN pip uninstall ray
+RUN pip install "numpy<2" -i https://pypi.tuna.tsinghua.edu.cn/simple
+RUN pip install ray==2.37.0
 RUN pip3 install -r requirements.txt -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com