![29c6fe3898b42545ef968ec76de3f973.png](https://img-blog.csdnimg.cn/img_convert/29c6fe3898b42545ef968ec76de3f973.png)
今天写了一个GPU排队脚本,事实上还是挺实用的。有的服务器是多用户使用,GPU的资源常常被占据着,很可能在夜间GPU空闲了,但来不及运行自己的脚本。如果没有和别人共享服务器的话,自己的多个程序想排队使用GPU,也可以用这个脚本。环境非常简单,有python就行了:
python==3.7
ubuntu==16.04
先创建脚本:
vim narrow_setup.py
cmd = 'python xxx.py' 这句可以设置你要运行的python脚本
# author: muzhan
直接运行脚本:
python narrow_setup.py
就可以监听nvidia-smi中的信息,以伺机触发python脚本~
运行结果如下:
![8ca1be23f9613c6875380e68e9dd700f.png](https://img-blog.csdnimg.cn/img_convert/8ca1be23f9613c6875380e68e9dd700f.png)
等待机会中... 如果gpu显存和功耗低于某个值时,就会触发python脚本。
文章首次发表于:
【猥琐发育】GPU排队脚本--一旦空闲就触发python脚本 - 木盏 - CSDN博客muzhan.blog.csdn.net![02235a669eafd7f1ad5d3b3a42c269e8.png](https://img-blog.csdnimg.cn/img_convert/02235a669eafd7f1ad5d3b3a42c269e8.png)