nvidiasmi.py 600 B

1234567891011121314151617181920212223
  1. import time
  2. import subprocess
  3. import locale
  4. import codecs
  5. import os
  6. cmd = "nvidia-smi"
  7. interval = 3
  8. while True:
  9. ps = subprocess.Popen(cmd, stdin=subprocess.PIPE, stdout=subprocess.PIPE, shell=True)
  10. for i in range(0,13):
  11. data = ps.stdout.readline()
  12. # bytes转化为str
  13. data = str(data)
  14. # 判断是否开始于"b'",去掉
  15. if data.startswith('b\''):
  16. data = data[2:]
  17. #同样去掉尾部
  18. if data.endswith('\\r\\n\''):
  19. data = data[:len(data)-5]
  20. print(data)
  21. print("\n\n\n\n\n\n\n")
  22. time.sleep(interval)