update to drive

This commit is contained in:
Your Name
2024-07-17 04:49:01 +00:00
parent f685bf6d80
commit adc45bc432
2 changed files with 54 additions and 24 deletions

View File

@@ -28,6 +28,9 @@ NCCL_IB_DISABLE=0
## test command
docker run -it --rm --network=host --privileged --ipc=host --ulimit memlock=-1 --gpus all hotwa/notebook:ngc
docker run --rm -it --gpus all --ipc=host --ulimit memlock=-1 --ulimit stack=67108864 hotwa/notebook:ngc /bin/bash
```shell
nvidia-smi
nvcc -V
@@ -40,7 +43,7 @@ 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
ofed_info -s # 如果输出显示了 OFED 版本号,则说明 OFED 驱动已安装。
mst version
mpirun --version
```