Commit e7481da7 authored by chenpangpang's avatar chenpangpang
Browse files

feat: 更新readme

parent d8df301a
......@@ -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`,输出:
```
......
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