Pytorch中查看GPU信息

本文摘自：知乎 用Pytorch中查看GPU信息

1. 返回当前设备索引

torch.cuda.current_device()

2. 返回GPU的数量

torch.cuda.device_count()

3. 返回gpu名字，设备索引默认从0开始

torch.cuda.get_device_name(0)

4. cuda是否可用

torch.cuda.is_available()