diy-smi 1.0(DIY System management interface)
在我们训练自己深度学习模型时,我们想看GPU信息,也想看CPU和内存信息。我常用的是nvidia-smi和htop,同时要开两个窗口,并且很多参数,我其实并不想查看。我们可以根据我们想要查看硬件那些信息,来自己diy一个监控窗口。
当然,你也可以在我的基础上,继续添加自己想要的元素,制作更炫酷的smi。
使用:
0️⃣ 先将代码下载,克隆,复制到本地。:Github diy-smi 1.0
1️⃣ 首先需要安装三个包,分别是用于收集gpu信息,cpu信息,美化我们的输出窗口。
pip install nvidia-ml-py3 -i https://pypi.mirrors.ustc.edu.cn/simple
pip install psutil -i https://pypi.mirrors.ustc.edu.cn/simple
pip install rich -i https://pypi.mirrors.ustc.edu.cn/simple
2️⃣ 然后运行我们的脚本就可以了
python diy-smi.py
3️⃣ 如果想用 watch -n 1命令,输出可能会没有颜色,在我电脑上没有颜色。
watch -c -n 1 python diy-smi.py --color