#cuda是否可用；
torch.cuda.is_available()# 返回gpu数量；
torch.cuda.device_count()# 返回gpu名字，设备索引默认从0开始；
torch.cuda.get_device_name(0)# 返回当前设备索引；
torch.cuda.current_device()
版权声明：本文为guaizaiguaizai原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
#cuda是否可用；
torch.cuda.is_available()# 返回gpu数量；
torch.cuda.device_count()# 返回gpu名字，设备索引默认从0开始；
torch.cuda.get_device_name(0)# 返回当前设备索引；
torch.cuda.current_device()