From a2597501cf463cfe887cc0b6f646bfe3b19502e7 Mon Sep 17 00:00:00 2001
From: lingyuzeng <pylyzeng@gmail.com>
Date: Mon, 15 Jul 2024 11:48:52 +0800
Subject: [PATCH] add test commmand

---
 finetune/README.md | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/finetune/README.md b/finetune/README.md
index f108437..ec3ad7f 100644
--- a/finetune/README.md
+++ b/finetune/README.md
@@ -5,6 +5,25 @@ docker-compose -f docker-compose_pytorch1.13.yml build
 docker-compose -f docker-compose_pytorch2.3.yml build
 ```
 
+## 镜像测试命令
+
+```shell
+nvidia-smi
+nvcc -V
+ninja --version
+ds_report
+python -c "import torch; print('torch:', torch.__version__, torch)"
+python -c "import torch; print('CUDA available:', torch.cuda.is_available())"
+python -c "import deepspeed; deepspeed.ops.op_builder.CPUAdamBuilder().load()"
+python -c "from flash_attn import flash_attn_func, flash_attn_varlen_func"
+python -c "import apex.amp; print('Apex is installed and the amp module is available.')"
+python -c "from xformers import ops as xops"
+ibstat
+ofed_info  -s
+mst version
+mpirun --version
+```
+
 ## 物理机更新内核
 
 ```shell