Commit dcab9349 authored by dcuai's avatar dcuai
Browse files

Update README.md

parent ef978b2a
...@@ -21,7 +21,7 @@ LinFusion: 1 GPU, 1 Minute, 16K Image ...@@ -21,7 +21,7 @@ LinFusion: 1 GPU, 1 Minute, 16K Image
推荐使用docker方式运行, 此处提供[光源](https://www.sourcefind.cn/#/service-details)拉取docker镜像的地址与使用步骤 推荐使用docker方式运行, 此处提供[光源](https://www.sourcefind.cn/#/service-details)拉取docker镜像的地址与使用步骤
``` ```
docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-ubuntu20.04-dtk24.04.2-py3.10 docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-ubuntu20.04-dtk24.04.2-py3.10
docker run -it --shm-size=1024G -v /path/your_code_data/:/path/your_code_data/ -v /opt/hyhal:/opt/hyhal --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name linfusion_pytorch <your IMAGE ID> bash # <your IMAGE ID>为以上拉取的docker的镜像ID替换,本镜像为:4555f389bc2a docker run -it --shm-size=1024G --network host -v /path/your_code_data/:/path/your_code_data/ -v /opt/hyhal:/opt/hyhal --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name linfusion_pytorch <your IMAGE ID> bash # <your IMAGE ID>为以上拉取的docker的镜像ID替换,本镜像为:4555f389bc2a
cd /path/your_code_data/ cd /path/your_code_data/
pip install git+https://github.com/openai/CLIP.git pip install git+https://github.com/openai/CLIP.git
pip install click clean-fid open_clip_torch pip install click clean-fid open_clip_torch
...@@ -87,7 +87,7 @@ bash ./examples/train/train.sh ...@@ -87,7 +87,7 @@ bash ./examples/train/train.sh
### 单机多卡 ### 单机多卡
``` ```
bash ./examples/training/distill.sh bash ./examples/train/distill.sh
``` ```
## 推理 ## 推理
......
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