写一个检测nvidia剩余显存的sh脚本
时间: 2024-03-27 10:35:53 浏览: 16
好的,下面是一个检测NVIDIA显存的sh脚本。
```
#!/bin/bash
# 检测NVIDIA显存的脚本
# 获取显卡数量
GPU_COUNT=$(nvidia-smi -L | wc -l)
# 循环遍历显卡
for ((i=0;i<$GPU_COUNT;i++)); do
# 获取显存使用情况
MEMORY_USAGE=$(nvidia-smi --id=$i --query-gpu=memory.used --format=csv,noheader)
# 获取显存总量
MEMORY_TOTAL=$(nvidia-smi --id=$i --query-gpu=memory.total --format=csv,noheader)
# 计算显存使用率
MEMORY_USED_PERCENT=$(echo "scale=2;$MEMORY_USAGE/$MEMORY_TOTAL*100" | bc)
# 输出显存使用情况
echo "GPU $i Memory Usage: $MEMORY_USAGE/$MEMORY_TOTAL MB ($MEMORY_USED_PERCENT%)"
done
```
使用方法:
1. 将上述脚本保存为`check_gpu_memory.sh`文件
2. 打开终端,执行以下命令:
```
chmod +x check_gpu_memory.sh
```
3. 执行以下命令来运行脚本:
```
./check_gpu_memory.sh
```
脚本将会输出每个显卡的显存使用情况,包括已使用的显存量、总显存量和使用率。