from paddle import fluid import os import multiprocessing gpu_dev_count = int(fluid.core.get_cuda_device_count()) cpu_dev_count = int(os.environ.get('CPU_NUM', multiprocessing.cpu_count())) dev_count = gpu_dev_count if gpu_dev_count > 0 else cpu_dev_count
