Update dtk24.04.1

README.md

@@ -31,9 +31,9 @@
 
 ### Docker
 
-    docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.13.1-centos7.6-dtk-23.04.1-py39-latest
+    docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-ubuntu20.04-dtk24.04.1-py3.10
 
-    docker run --shm-size 10g --network=host --name=codeformer --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v 项目地址(绝对路径):/home/ -it <your IMAGE ID> bash
+    docker run --shm-size 10g --network=host --name=codeformer --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v /opt/hyhal:/opt/hyhal -v 项目地址(绝对路径):/home/ -it <your IMAGE ID> bash
 
     pip install -r requirements.txt
 
@@ -42,11 +42,7 @@
     python basicsr/setup.py develop
 
     # 以下内容可选，仅在视频增强时需要
-    yum install epel-release -y
-
-    yum localinstall --nogpgcheck https://download1.rpmfusion.org/free/el/rpmfusion-free-release-7.noarch.rpm -y
-
-    yum install ffmpeg ffmpeg-devel -y
+    apt update && apt install ffmpeg
 
 ### Dockerfile
 
@@ -54,7 +50,7 @@
     docker build -t <IMAGE_NAME>:<TAG> .
 
     # <your IMAGE ID>用以上拉取的docker的镜像ID替换
-    docker run -it --shm-size 10g --network=host --name=codeformer --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined <your IMAGE ID> bash
+    docker run -it --shm-size 10g --network=host --name=codeformer --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v /opt/hyhal:/opt/hyhal  <your IMAGE ID> bash
 
     pip install -r requirements.txt
 
@@ -63,11 +59,7 @@
     python basicsr/setup.py develop
 
     # 以下内容可选，仅在视频增强时需要
-    yum install epel-release -y
-
-    yum localinstall --nogpgcheck https://download1.rpmfusion.org/free/el/rpmfusion-free-release-7.noarch.rpm -y
-
-    yum install ffmpeg ffmpeg-devel -y
+    apt update && apt install ffmpeg
     
 
 ## 数据集
@@ -111,7 +103,7 @@
     python -m torch.distributed.launch --nproc_per_node=8 --master_port=4323 basicsr/train.py -opt options/CodeFormer_stage3.yml --launcher pytorch
 
 
-注意：`vqgan`为`net_g_xxx.pth`，`net_d`为第一阶段生成。
+注意：`vqgan`为`net_g_xxx.pth`为第二阶段生成，`net_d`为第一阶段生成。
 
 ## 推理