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