查看显卡名称 pytorch

import torch

torch.cuda.get_device_name()  #查看自己的显卡

【推广】 免费学中医，健康全家人
原文地址：https://www.cnblogs.com/gagaein/p/14352856.html