更新dtk24.04.1镜像

README.md

@@ -23,8 +23,8 @@ mv itransformer_pytorch iTransformer # 去框架名后缀
 
 ### Docker（方法一）
 ```
-docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-centos7.6-dtk23.10-py38
-# <your IMAGE ID>为以上拉取的docker的镜像ID替换，本镜像为：ffa1f63239fc
+docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-ubuntu20.04-dtk24.04.1-py3.10
+# <your IMAGE ID>为以上拉取的docker的镜像ID替换
 docker run -it --shm-size=32G -v $PWD/iTransformer:/home/iTransformer -v /opt/hyhal:/opt/hyhal --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name itransformer <your IMAGE ID> bash
 cd /home/iTransformer
 pip install -r requirements.txt # requirements.txt
@@ -40,8 +40,8 @@ docker run --shm-size=32G --name itransformer -v /opt/hyhal:/opt/hyhal:ro --priv
 1、关于本项目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
 ```