Skip to content

GitLab

Commit 049f76e2 authored by lijian6's avatar lijian6
Browse files

change docker which has dtk

parent 25e2b8a5
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 13 deletions
README.md
View file @ 049f76e2
...@@ -17,11 +17,12 @@ ...@@ -17,11 +17,12 @@
### Docker（方法一）： ### Docker（方法一）：
``` ```
docker pull image.sourcefind.cn:5000/dcu/admin/base/custom:stablediffusion_v2-1 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 bash 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（方法二）： ### Dockerfile（方法二）：
``` ```
cd docker
docker build --no-cache -t stablediffussion:test . 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 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 ...@@ -32,16 +33,6 @@ cd /stablediffusion-main
git lfs install git lfs install
git clone https://huggingface.co/stabilityai/stable-diffusion-2-1-base 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
View file @ 049f76e2
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
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment