Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
LinFusion_pytorch
Commits
dcab9349
Commit
dcab9349
authored
Oct 30, 2024
by
dcuai
Browse files
Update README.md
parent
ef978b2a
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
2 deletions
+2
-2
README.md
README.md
+2
-2
No files found.
README.md
View file @
dcab9349
...
@@ -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/train
ing
/distill.sh
bash ./examples/train/distill.sh
```
```
## 推理
## 推理
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment