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

Question

我可以通过 mxnet 查看可用的 GPU 吗？

tensorflow 有没有类似的东西

tf.test.gpu_device_name()

在 mxnet 中？

Answer 1

检查您的 GPU 是否正在被使用的 final方法是使用 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)

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

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