https://github.com/PeizhuoLi/ganimator
Tip revision: 2943064b456d13f0357e23e3c37fb43b6aa3fdaa authored by Peizhuo Li on 09 September 2022, 14:56:35 UTC
Update README.md
Update README.md
Tip revision: 2943064
utils.py
def get_gpu_info():
try:
import GPUtil
gpus = GPUtil.getGPUs()
return gpus[0].name if len(gpus) > 0 else 'NO GPU detected'
except:
return 'NO GPUtil installed'
def get_cpu_info():
try:
import cpuinfo
info = cpuinfo.get_cpu_info()
return info['brand_raw']
except:
return 'NO cpuinfo installed'
def get_device_info():
return get_cpu_info(), get_gpu_info()