PyTorch学习系列(十六)——如何使用cuda进行训练？

如果想在CUDA上进行计算，需要将操作对象放在GPU内存中。
对于普通的张量，可以直接：

x = torch.randn(2, 3)
x = x.cuda()

对于神经网络：

model = MyModel()
model.cuda()

同一个GPU上的张量计算结果仍然保存在该GPU上。

参考

[1] http://pytorch.org/docs/notes/cuda.html#