2. Pytorch使用GPU加速. Pytorch中使用GPU加速模型非常简单,只要将模型和数据移动到GPU上。核心代码只有以下几行。 1# 定义模型 2device = torch.device("cuda:0" if ...
確定! 回上一頁