PyTorch如何获得显卡 Compute Compatibility
前言什么是Compute compatibility(下文简称CC)?它是NVIDIA为各代显卡(包括Jetson)设定的一个值,和算力没关系,直译过来叫“计算兼容性”,通常更高的CC可以跑更多的网络结构,硬编码质量也更高。正文老规矩,直接上代码:import torchcc = torch.cuda.get_device_capability()cc = f'{cc[0]}{cc[1]}'print(cc)虽然很简单,还是解读一下吧torch.cuda.get_devic
原创
2021-01-01 22:31:15 ·
1179 阅读 ·
0 评论