使用 GPU

中文版 pytorch
#1

http://zh-v2.d2l.ai/chapter_deep-learning-computation/use-gpu.html

#2

torch.cuda.device可以被用来做什么呢？看起来在gpu上操作变量只需要torch.device就可以了。

#3

以及5.6.3.中的代码，不需要写net = net.to(try_gpu())吧？直接net.to(try_gpu())就可以了。测试看来.to对于nn.Module是in-place操作。