PyTorch 非常容易的就可以使用GPU,可以用如下方式把一个模型放到GPU上: device = torch.device("cuda: 0") model.to(device) Copy. 然后可以复制所有的张量到GPU上:.
確定! 回上一頁