有没有办法检查mxnet是否使用我的gpu？

Question

我能看到mxnet可用的gpus是什么吗？

tensorflow是否有类似的东西

tf.test.gpu_device_name()

在mxnet？

Answer 1

检查GPU是否被利用的最终方法是使用nvidia-smi命令。我最喜欢的论点是：

nvidia-smi --query-gpu=timestamp,name,pci.bus_id,driver_version,pstate,pcie.link.gen.max,pcie.link.gen.current,temperature.gpu,utilization.gpu,utilization.memory,memory.total,memory.free,memory.used --format=csv -l 1

如果您只想测试gpu支持是否可用（这是tf.test.gpu_device_name（）），则以下函数可以提供帮助：

import mxnet as mx 
def gpu_device(gpu_number=0):
    try:
        _ = mx.nd.array([1, 2, 3], ctx=mx.gpu(gpu_number))
    except mx.MXNetError:
        return None
    return mx.gpu(gpu_number)

如果请求的gpu设备不可用，则此函数返回None;如果gpu设备可用，则返回相关的上下文。您还可以使用此功能检查此系统上是否支持GPU：

if not gpu_device():
    print('No GPU device found!')