**pytorch查看GPU信息**
windows使用nvidia-smi查看gpu信息
python命令查看GPu信息
import torch
print(torch.cuda.is_available()) # 查看GPu设备是否可用
print(torch.cuda.device_count()) # 查看GPu设备数量
print(torch.cuda.get_device_name()) # 查看当前GPu设备名称，默认设备id从0开始
print(torch.cuda.current_device()) # 查看当前GPu设备id
版权声明：本文为qq_41694461原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接和本声明。