update README.md and dockerfile

README.md

@@ -36,13 +36,13 @@ python -m pip install -e detectron2
 -v 路径、docker_name和imageID根据实际情况修改
 
 ```
-docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.13.1-centos7.6-dtk-23.04-py38-latest
+docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-ubuntu20.04-dtk24.04.1-py3.8
 
-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 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 /home/deepsolo_pytorch
 pip install --upgrade setuptools wheel
-pip install -r requirements.txt
+pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
 python -m pip install 'git+https://github.com/facebookresearch/detectron2.git'
 bash make.sh
 ```
@@ -55,11 +55,11 @@ bash make.sh
 cd ./docker
 
 docker build --no-cache -t deepsolo: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 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 /your_code_path/deepsolo_pytorch
 pip install --upgrade setuptools wheel
-pip install -r requirements.txt
+pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
 python -m pip install 'git+https://github.com/facebookresearch/detectron2.git'
 bash make.sh
 ```
@@ -69,10 +69,10 @@ bash make.sh
 1、关于本项目DCU显卡所需的特殊深度学习库可从光合开发者社区下载安装: https://developer.hpccube.com/tool/
 
 ```
-DTK软件栈：dtk23.04
+DTK软件栈：dtk24.04.1
 python：python3.8
-torch：1.13.1
-torchvision：0.14.1
+torch：2.1.0
+torchvision：0.16.0+das1.1.git7d45932.abi1.dtk2404.torch2.1
 ```
 
 Tips：以上dtk软件栈、python、torch等DCU相关工具版本需要严格一一对应
@@ -81,7 +81,7 @@ Tips：以上dtk软件栈、python、torch等DCU相关工具版本需要严格
 
 ```
 pip install --upgrade setuptools wheel
-pip install -r requirements.txt
+pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
 python -m pip install 'git+https://github.com/facebookresearch/detectron2.git'
 bash make.sh
 ```

docker/Dockerfile

-FROM image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.13.1-centos7.6-dtk-23.04-py38-latest
+FROM image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-ubuntu20.04-dtk24.04.1-py3.8