方法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()
import mxnet as mx
mx.test_utils.list_gpus()
# 或
mx.util.get_gpu_count()
pip install nvgpu
import nvgpu
nvgpu.available_gpus()