Pytorch 中使用torch.device()选取并返回抽象出的设备,然后在定义的网络模块或者Tensor后面加上.to(device变量)就可以将它们搬到设备上了。以上一篇代码为例,使用GPU ...
確定! 回上一頁