pytorch获取当前GPU信息

GTX_AI

pytorch获取当前GPU信息：

1. import torch
   
2. gpu_num = torch.cuda.device_count()
   
3. print('GPU数量：%d' % (gpu_num))
   
4. for i in range(gpu_num):
   
5.     gpu_name = torch.cuda.get_device_name(i)
   
6.     print('GPU名称：%s' % (gpu_name))
   
7.     gpu_busy = torch.cuda.is_available()
   
8.     print('%s是否可用：%d' % (gpu_name, gpu_busy))
   
9.     gpu_index = torch.cuda.current_device()
   
10.     print('当前GPU索引：%d' % (gpu_index))

复制代码