【推荐】很好用的小工具pynvml

pynvml是用于英伟达GPU管理和监视功能的Python接口。可以用它来开发一些pytorch或者tensorflow 监控工具。具体的可以看官网：
https://pythonhosted.org/nvidia-ml-py/
后续小工具写好后也会发出来。
可以通过源码也可以通pip安装。
pip安装过程：

pip install nvidia-ml-py

这时候一般就可以用了，如果出现一下报错：

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/home/GZKP/.conda/envs/py36/lib/python3.6/site-packages/pynvml.py", line 1831
    print c_count.value
                ^
SyntaxError: Missing parentheses in call to 'print'. Did you mean print(c_count.value)?

是因为安装的与python版本不匹配，重新显式指定版本安装一下就好了：

pip install nvidia-ml-py2 #python2

pip install nvidia-ml-py3 #python3

具体使用细节，官网上有Demo很好上手，这里就不写了。