怎么用 pytorch 查看 GPU 信息,怎么查看keras 或者 tensorflow 正在使用的GPU

2024年5月31日 190次阅读来源: pytorch

如果你用的 Keras 或者 TensorFlow, 请移步怎么查看keras 或者 tensorflow 正在使用的GPU

In [1]: import torch In [2]: torch.cuda.current_device() Out[2]: 0 In [3]: torch.cuda.device(0) Out[3]: <torch.cuda.device at 0x7efce0b03be0> In [4]: torch.cuda.device_count() Out[4]: 1 In [5]: torch.cuda.get_device_name(0) Out[5]: 'GeForce GTX 950M' In [6]: torch.cuda.is_available() Out[6]: True

    原文作者：pytorch
    原文地址: https://www.cnblogs.com/mashuai-191/p/11122353.html
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。