可爱小懒虫
Pytorch 使用GPU加速
Pytorch中使用torch.device()选取并返回抽象出的设备，然后在定义的网络模块或者Tensor后面加上.to(device变量)就可以将它们搬到设备上了。以上一篇代码为例，使用GPU设备：device = torch.device('cuda:0') # 使用第一张显卡需要将如下部分搬移到GPU上：1. 定义的网络net = MLP().to(device)...
他的专栏
他的归档
热门文章
