摘要:   在本节中,我们将学习如何利用 使用多个GPU。 在PyTorch中使用多个GPU非常容易,你可以使用下面代码将模型放在GPU上: 然后,你可以将所有张量拷贝到GPU上: 请注意,仅仅调用 并不会将张量拷贝到GPU上,你需要将它指派给一个新的张量,然后在GPU上使用这个新张量。 在多个G 阅读全文
posted @ 2018-01-31 10:34 HackHan 阅读(607) 评论(0) 推荐(0) 编辑
摘要:   太棒啦!到目前为止,你已经了解了如何定义神经网络、计算损失,以及更新网络权重。不过,现在你可能会思考以下几个方面: 0x01 数据集 通常,当你需要处理图像、文本、音频或视频数据时,你可以使用标准的python包将数据加载到numpy数组中。然后你可以将该数组转换成一个 。 对于图像, 阅读全文
posted @ 2018-01-31 01:25 HackHan 阅读(4027) 评论(1) 推荐(1) 编辑