Pytorch检查cuda

1. 是否可以用显卡：

torch.cuda.is_available()

2. 显卡个数：

torch.cuda.device_count()

3. 显卡名字：

torch.cuda.get_device_name(0)

黄世宇/Shiyu Huang's Personal Page：​​https://huangshiyu13.github.io/​​