#!/bin/sh
# 查看上一次关机日期和时间
# last -x|grep shutdown | head -1
# 查看最近一条重启记录
# last reboot | head -1
# 开机自启
# chmod +x /etc/rc.d/rc.local
#使用远程执行脚本，方便中断。需要安装curl
# echo "/usr/bin/curl http://192.168.1.34/reboot_gpu.sh | /usr/bin/sh" >> /etc/rc.d/rc.local

dir=/var/log/times

# echo 100 > $dir    #-f 参数判断 $file 是否存在
if [ ! -f "$dir" ]; then
  echo 10 > /var/log/times
  dir=/var/log/times
fi

i=`cat $dir`
if [ $i = 0 ]; then
  echo "删除启动项： vi /etc/rc.d/rc.local
  rm -rf $dir
  rm -rf ~/readme2clean" > ~/readme2clean
else
  i=`expr $i - 1`
  echo $i > $dir
  #运行gpuburn 600s，记录nvidia-smi
  nvidia-smi --query-gpu=timestamp,temperature.gpu,name,pci.bus_id,driver_version,pstate,pcie.link.gen.max,pcie.link.gen.current,pcie.link.width.max,pcie.link.width.current,utilization.gpu,utilization.memory,memory.total,memory.free,memory.used --format=csv -l 60 > /root/nvidia-smi$(date +%Y%m%d_%H%M%S).csv & cd /root/gpu-burn-master
  ./gpu_burn -tc 600
  sleep 61

  reboot
fi