add test commmand
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user