网站建设好发信息网,猪八戒托管赏金做网站,怎样设计网站,公司logo设计理念其原理是通过nvidia-smi扫描每块显卡上的内存#xff0c;然后查询已经使用的内存#xff0c;若已经使用的显卡内存不大于一个阈值#xff0c;则运行python脚本#xff08;你也可以替换成别的命令#xff09;
#!/bin/bash
var0
ocp_memory${2:-500}
while [ $var -eq 0 ]
…其原理是通过nvidia-smi扫描每块显卡上的内存然后查询已经使用的内存若已经使用的显卡内存不大于一个阈值则运行python脚本你也可以替换成别的命令
#!/bin/bash
var0
ocp_memory${2:-500}
while [ $var -eq 0 ]
echo waiting for available gpu...
docount0for i in $(nvidia-smi --query-gpumemory.used --formatcsv,noheader,nounits)doif [ $i -lt ocp_memory ]thenecho GPU$count is avaiableCUDA_VISIBLE_DEVICES$count python ${1}var1breakficount$(($count1)) done
done使用时的命令为
sh ocp_gpu.sh test.py 500test.py即需要运行的py文件500代表显卡被占有内存小于等于500MB。
