PROSAGA码农传奇-深度学习-有没有办法检查mxnet是否使用我的gpu？

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

作者: afs-loliaholic
发布时间: 2024-12-27 05:22:45 (17天前)
转自：

<div class =“post-text”itemprop =“text”>
  <P>
    检查GPU是否被利用的最终方法是使用
    <a href="http://developer.download.nvidia.com/compute/DCGM/docs/nvidia-smi-367.38.pdf" rel="noreferrer">
       <code>
        nvidia-smi
      </code>
    </A>
     命令。我最喜欢的论点是：
  </p>
   <pre>
    <code>
      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

</code>
  </pre>
  <P>
    如果您只想测试gpu支持是否可用（这是tf.test.gpu_device_name（）），则以下函数可以提供帮助：
  </p>
   <pre>
    <code>
      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)

</code>
  </pre>
  <P>
    此函数返回
     <code>
      None
    </code>
     如果请求的gpu设备不可用，或者如果gpu设备可用则返回相关的上下文。您还可以使用此功能检查此系统上是否支持GPU：
  </p>
   <pre>
    <code>
      if not gpu_device():
    print('No GPU device found!')

</code>
  </pre>
</DIV>