[pytorch]查看GPU信息

20181128 qzd

https://pytorch.org/docs/stable/cuda.html

1. 查看GPU是否可用

torch.cuda.is_available()

2. 返回GPU数量

torch.cuda.device_count()

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

torch.cuda.get_device_name(0)

4. 返回当前设备索引

torch.cuda.current_device()