Commit e7481da7 authored by chenpangpang's avatar chenpangpang
Browse files

feat: 更新readme

parent d8df301a
...@@ -7,7 +7,7 @@ ...@@ -7,7 +7,7 @@
1. 下载代码:`git clone http://developer.hpccube.com/codes/chenpangpang/gpu-base-image-test.git` 1. 下载代码:`git clone http://developer.hpccube.com/codes/chenpangpang/gpu-base-image-test.git`
2. 下载模型: `cd gpu-base-image-test && python hf_down.py` 2. 下载模型: `cd gpu-base-image-test && python hf_down.py`
3. 确认要构建的镜像 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)构建镜像 - 基于[pytorch官方镜像](https://hub.docker.com/r/pytorch/pytorch)构建镜像
```bash ```bash
...@@ -72,7 +72,7 @@ torch cudnn version: 8902 ...@@ -72,7 +72,7 @@ torch cudnn version: 8902
torchvision version: 0.18.1 torchvision version: 0.18.1
torchaudio version: 2.3.1 torchaudio version: 2.3.1
``` ```
确认`输出的版本信息``镜像名称`是否匹配,确认`torch cuda`是否可用。 确认`输出的版本信息``镜像名称`是否匹配,确认`torch cuda`是否可用。<br>
2. 文本生成验证:运行:`sh script/2_text_generate_test.sh $IMAGE_NAME`,输出: 2. 文本生成验证:运行:`sh script/2_text_generate_test.sh $IMAGE_NAME`,输出:
``` ```
Setting `pad_token_id` to `eos_token_id`:50256 for open-end generation. Setting `pad_token_id` to `eos_token_id`:50256 for open-end generation.
...@@ -80,7 +80,7 @@ torchaudio version: 2.3.1 ...@@ -80,7 +80,7 @@ torchaudio version: 2.3.1
"Let's start an internal test now, and then "Let's start an internal test now, and then
``` ```
确认`输出信息`是否符合预期。 确认`输出信息`是否符合预期。<br>
3. 图像生成验证:运行`sh script/3_image_generate_test.sh $IMAGE_NAME`,输出: 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