更换dtk24.04.1镜像

README.md

@@ -23,7 +23,7 @@ EfficientSAM模型利用掩码图像预训练（SAMI），该预训练学习从S
 ### Docker（方法一）
 此处提供[光源](https://www.sourcefind.cn/#/service-details)拉取docker镜像的地址与使用步骤
 ```
-docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-centos7.6-dtk23.10-py38
+docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-ubuntu20.04-dtk24.04.1-py3.10
 docker run -it --shm-size=64G -v /path/your_code_data/:/path/your_code_data/ -v /opt/hyhal:/opt/hyhal --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name efficientsam_pytorch  <your IMAGE ID> bash # <your IMAGE ID>为以上拉取的docker的镜像ID替换，本镜像为：ffa1f63239fc
 cd /path/your_code_data/efficientsam_pytorch
 pip install -e .
@@ -36,7 +36,7 @@ pip install -e .
 此处提供dockerfile的使用方法
 ```
 docker build --no-cache -t efficientsam:latest .
-docker run -it --shm-size=64G -v /path/your_code_data/:/path/your_code_data/ -v /opt/hyhal:/opt/hyhal --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name efficientsam_pytorch  efficientsam  bash
+docker run -it --shm-size=64G -v /path/your_code_data/:/path/your_code_data/ -v /opt/hyhal:/opt/hyhal --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name efficientsam_pytorch  efficientsam:latest  bash
 cd /path/your_code_data/efficientsam_pytorch
 pip install -e .
 pip install -r requirements.txt -i https://mirrors.aliyun.com/pypi/simple/  --trusted-host mirrors.aliyun.com
@@ -49,8 +49,8 @@ pip install -e .
 
 关于本项目DCU显卡所需的特殊深度学习库可从[光合](https://developer.hpccube.com/tool/)开发者社区下载安装。
 ```
-DTK驱动：dtk23.10
-python：python3.8
+DTK驱动：dtk24.04.1
+python：python3.10
 torch: 2.1.0
 torchvision: 0.16.0
 triton:2.1.0