"git@developer.sourcefind.cn:OpenDAS/fairscale.git" did not exist on "8a05ff76a70485b69719078fb78402ad5874760d"
Commit 1739e35c authored by wangwei990215's avatar wangwei990215
Browse files

Update README.md

parent 5458923b
...@@ -15,7 +15,7 @@ ...@@ -15,7 +15,7 @@
## 环境配置 ## 环境配置
### Docker: ### Docker(方法一)
``` ```
docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-py3.8-dtk24.04.3-ubuntu20.04 docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-py3.8-dtk24.04.3-ubuntu20.04
docker run --shm-size 10g --network=host --name=stablediffusion_v2-1 --privileged --device=/dev/kfd --device=/dev/mkfd --device=/dev/dri -v /opt/hyhal:/opt/hyhal --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -it image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-py3.8-dtk24.04.3-ubuntu20.04 bash docker run --shm-size 10g --network=host --name=stablediffusion_v2-1 --privileged --device=/dev/kfd --device=/dev/mkfd --device=/dev/dri -v /opt/hyhal:/opt/hyhal --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -it image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-py3.8-dtk24.04.3-ubuntu20.04 bash
...@@ -26,6 +26,17 @@ cd examples/text_to_image ...@@ -26,6 +26,17 @@ cd examples/text_to_image
pip install -r requirements.txt pip install -r requirements.txt
``` ```
### Anaconda(方法三):
此处提供本地配置、编译的详细步骤,例如:<br>
关于本项目DCU显卡所需的特殊深度学习库可从[光合开发者社区](https://developer.sourcefind.cn/)下载安装
```
DTK:dtk24.04.3
python:3.10
torch:2.1.0
diffusers和其他深度学习库(requirements)可以参考方法一进行安装。
```
tips:以上dtk驱动、python、torch等DCU相关工具需要严格一一对应。
## 数据集 ## 数据集
pokemon-blip-captions <br> pokemon-blip-captions <br>
可以从以下网址下载: 可以从以下网址下载:
......
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