获取cpu的个数
import os
def get_cpus():
if 'NUMBER_OF_PROCESSORS' in os.environ:
cpu_num = os.environ['NUMBER_OF_PROCESSORS']
return cpu_num
return 0
获取gpu的个数
nvidia_path = "C:\\Program Files\\NVIDIA Corporation\\NVSMI\\nvidia-smi.exe"
def get_gpus(nvidia_path):
if not os.path.exists(nvidia_path):
logging.error("Can not find nvidia smi exe, please check the path.")
return 0
cmd = '"{0}" -L'.format(nvidia_path)
gpus = 0
try:
with os.popen(cmd) as fd:
for line in fd:
if 'GPU' in line:
gpus += 1
except Exception as e:
logging.error("Get gpu count failed with error {0}".format(e))
return gpus
这里包含了带空格的cmd命令怎么处理, 带空格的路径先用“” 引起来, 然后结合参数再在外面包一层单引号.
