查看 PyTorch 是否使用 GPU

借助 torch.cuda 中的命令，可以帮助我们查看 PyTorch 是否使用 GPU。

Python

import torch

当前设备索引

Python

torch.cuda.current_device()

可用的 GPU 数量

Python

torch.cuda.device_count()

GPU 设备名字，设备索引默认从 0 开始

Python

torch.cuda.get_device_name(0)

检查 CUDA 是否可用

Python

torch.cuda.is_available()

评论