扫二维码与项目经理沟通
我们在微信上24小时期待你的声音
解答本文疑问/技术咨询/运营咨询/技术建议/互联网交流
其原理是通过nvidia-smi扫描每块显卡上的内存，然后查询已经使用的内存，若已经使用的显卡内存不大于一个阈值，则运行python脚本（你也可以替换成别的命令）
#!/bin/bash
var=0
ocp_memory=${2:-500}
while [ $var -eq 0 ]
echo 'waiting for available gpu...'
do
count=0
for i in $(nvidia-smi --query-gpu=memory.used --format=csv,noheader,nounits)
do
if [ $i -lt ocp_memory ]
then
echo 'GPU'$count' is avaiable'
CUDA_VISIBLE_DEVICES=$count python ${1}
var=1
break
fi
count=$(($count+1))
done
done使用时的命令为
sh ocp_gpu.sh test.py 500test.py即需要运行的py文件，500代表显卡被占有内存小于等于500MB。
你是否还在寻找稳定的海外服务器提供商？创新互联www.cdcxhl.cn海外机房具备T级流量清洗系统配攻击溯源，准确流量调度确保服务器高可用性，企业级服务器适合批量采购，新人活动首月15元起，快前往官网查看详情吧
我们在微信上24小时期待你的声音
解答本文疑问/技术咨询/运营咨询/技术建议/互联网交流
微信二维码
移动版官网