查看显卡名称 pytorch

import torch

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

 

posted @ 2021-01-31 16:05  乌蝇哥  阅读(241)  评论(0)  编辑  收藏  举报