update README.md

README.md

@@ -23,7 +23,7 @@ DETR将目标检测看作一种set prediction问题, 并提出了一个十分简
 ### Docker（方法一）
 
 ```
-docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-centos7.6-dtk24.04-py310
+docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-ubuntu20.04-dtk24.04.1-py3.10 
 
 docker run -it -v /path/your_code_data/:/path/your_code_data/ -v /opt/hyhal:/opt/hyhal:ro --shm-size=32G --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name docker_name imageID bash
 
@@ -47,10 +47,10 @@ pip install -r requirements.txt
 1、关于本项目DCU显卡所需的特殊深度学习库可从光合开发者社区下载安装: https://developer.hpccube.com/tool/
 
 ```
-DTK软件栈: dtk24.04
+DTK软件栈: dtk24.04.1
 python: python3.10
-torch: 2.1.0+git00661e0.abi0.dtk2404
-torchvision: 0.16.0+gitc9e7141.abi0.dtk2404.torch2.1
+torch: 2.1.0+das1.1.git3ac1bdd.abi1.dtk2404
+torchvision: 0.16.0+das1.1.git7d45932.abi1.dtk2404.torch2.1
 ```
 
 Tips: 以上dtk软件栈、python、torch等DCU相关工具版本需要严格一一对应

docker/Dockerfile

-FROM image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-centos7.6-dtk24.04-py310
+FROM image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-ubuntu20.04-dtk24.04.1-py3.10 

requirements.txt

 cython
 pycocotools==2.0.7
 submitit==1.5.1
-torch>=1.5.0
-torchvision>=0.6.0
+# torch>=1.5.0
+# torchvision>=0.6.0
 scipy
 onnx
 onnxruntime
-numpy
+numpy==1.26.4
\ No newline at end of file