Update README.md

README.md

@@ -14,8 +14,8 @@ Centernet对于输入图像(512,512,3)，下采样率为4，CenterNet将得到
 ### Docker（方法一）
 
 ```
-docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.13.1-centos7.6-dtk-23.04.1-py38-latest
-docker run -it -v /path/your_code_data/:/path/your_code_data/ --shm-size=32G --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name docker_name imageID bash
+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
 cd /path/workspace/
 pip install mmdet==3.2.0 -i https://mirrors.aliyun.com/pypi/simple/
 ```
@@ -29,11 +29,11 @@ cd /path/workspace/
 ### Anaconda（方法三）
 1、关于本项目DCU显卡所需的特殊深度学习库可从光合开发者社区下载安装： https://developer.hpccube.com/tool/
 ```
-DTK软件栈: dtk23.04.1
-python: python3.8
-torch: 1.13.1
-torchvision: 0.14.1
-mmcv: 2.0.0
+DTK软件栈: dtk24.04.1
+python: python3.10
+torch: 2.1.0
+torchvision: 0.16.0
+mmcv: 2.0.1
 
 ```