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
chenpangpang
gpu-base-image-build
Commits
e7481da7
Commit
e7481da7
authored
Sep 19, 2024
by
chenpangpang
Browse files
feat: 更新readme
parent
d8df301a
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
3 deletions
+3
-3
README.md
README.md
+3
-3
No files found.
README.md
View file @
e7481da7
...
...
@@ -7,7 +7,7 @@
1.
下载代码:
`git clone http://developer.hpccube.com/codes/chenpangpang/gpu-base-image-test.git`
2.
下载模型:
`cd gpu-base-image-test && python hf_down.py`
3.
确认要构建的镜像
-
镜像制作进度:https://bvjoh3z2qoz.feishu.cn/base/BKl6birVbarmzJsnznkcEDFTnV9?table=tbl3bCdS7qfjPn6j&view=vewww0URg8
-
镜像制作进度:https://bvjoh3z2qoz.feishu.cn/base/BKl6birVbarmzJsnznkcEDFTnV9?table=tbl3bCdS7qfjPn6j&view=vewww0URg8
## 镜像构建
-
基于
[
pytorch官方镜像
](
https://hub.docker.com/r/pytorch/pytorch
)
构建镜像
```
bash
...
...
@@ -72,7 +72,7 @@ torch cudnn version: 8902
torchvision version: 0.18.1
torchaudio version: 2.3.1
```
确认
`输出的版本信息`
和
`镜像名称`
是否匹配,确认
`torch cuda`
是否可用。
确认
`输出的版本信息`
和
`镜像名称`
是否匹配,确认
`torch cuda`
是否可用。
<br>
2.
文本生成验证:运行:
`sh script/2_text_generate_test.sh $IMAGE_NAME`
,输出:
```
Setting `pad_token_id` to `eos_token_id`:50256 for open-end generation.
...
...
@@ -80,7 +80,7 @@ torchaudio version: 2.3.1
"Let's start an internal test now, and then
```
确认
`输出信息`
是否符合预期。
确认
`输出信息`
是否符合预期。
<br>
3.
图像生成验证:运行
`sh script/3_image_generate_test.sh $IMAGE_NAME`
,输出:
```
...
...
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