python获取GPU数量

方法1：mxnet

import mxnet as mx
mx.test_utils.list_gpus()
# 或
mx.util.get_gpu_count()

方法2： nvgpu

pip install nvgpu

import nvgpu
nvgpu.available_gpus()