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
|
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
|
```shell
|
||||||
|
|||||||
Reference in New Issue
Block a user