change docker which has dtk

README.md

@@ -17,11 +17,12 @@
 
 ### Docker（方法一）：
 ```
-docker pull image.sourcefind.cn:5000/dcu/admin/base/custom:stablediffusion_v2-1
-docker run --rm --shm-size 10g --network=host --name=stablediffussion_v2-1 --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -it image.sourcefind.cn:5000/dcu/admin/base/custom:stablediffusion_v2-1 bash
+docker pull image.sourcefind.cn:5000/dcu/admin/base/custom:stablediffusion_v2-1_dtk24.04_xformers0.0.25_py310
+docker run --rm --shm-size 10g --network=host --name=stablediffussion_v2-1 --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -it image.sourcefind.cn:5000/dcu/admin/base/custom:stablediffusion_v2-1_dtk24.04_xformers0.0.25_py310 bash
 ```
 ### Dockerfile（方法二）：
 ```
+cd docker
 docker build --no-cache -t stablediffussion:test .
 docker run --rm --shm-size 10g --network=host --name=stablediffussion_v2-1 --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -it stablediffussion:test bash
 ```
@@ -32,16 +33,6 @@ cd /stablediffusion-main
 git lfs install
 git clone https://huggingface.co/stabilityai/stable-diffusion-2-1-base
 ```
-### 下载 dtk工具
-```
-cd /opt/
-## 下载 dtk
-https://cancon.hpccube.com:65024/directlink/1/DTK-24.04/CentOS7.6/DTK-24.04-CentOS7.6-x86_64.tar.gz
-解压链接
-tar -zxvf DTK-24.04-CentOS7.6-x86_64.tar.gz
-ln -s DTK-24.04 dtk
-```
-
 ## 数据集
 无
 

docker/Dockerfile

-FROM image.sourcefind.cn:5000/dcu/admin/base/custom:stablediffusion
+FROM image.sourcefind.cn:5000/dcu/admin/base/custom:stablediffusion_v2-1_dtk24.04_xformers0.0.25_py310