摘要:
译者:bat67 最新版会在译者仓库首先同步。 在这个教程里,我们将学习如何使用数据并行(DataParallel)来使用多GPU。 PyTorch非常容易的就可以使用GPU,可以用如下方式把一个模型放到GPU上: device = torch.device("cuda:0") model.to(d 阅读全文
posted @ 2019-01-25 20:48
绝不原创的飞龙
阅读(28)
评论(0)
推荐(0)
摘要:
译者:bat67 最新版会在译者仓库首先同步。 可以使用torch.nn包来构建神经网络. 我们以及介绍了autograd,nn包依赖于autograd包来定义模型并对它们求导。一个nn.Module包含各个层和一个forward(input)方法,该方法返回output。 例如,下面这个神经网络可 阅读全文
posted @ 2019-01-25 20:45
绝不原创的飞龙
阅读(33)
评论(0)
推荐(0)
摘要:
译者:bat67 最新版会在译者仓库首先同步。 PyTorch是一个基于python的科学计算包,主要针对两类人群: 作为NumPy的替代品,可以利用GPU的性能进行计算作为一个高灵活性、速度快的深度学习平台 入门 张量 Tensor(张量)类似于NumPy的ndarray,但还可以在GPU上使用来 阅读全文
posted @ 2019-01-25 20:43
绝不原创的飞龙
阅读(25)
评论(0)
推荐(0)

浙公网安备 33010602011771号