Training a Classifier, PyTorch Core Team, 2024 (PyTorch) - 一个官方PyTorch教程,实际演示了如何在训练工作流中使用torch.device和.to()将模型和输入数据移动到GPU。
Deep Learning with PyTorch, Eli Stevens, Luca Antiga, Thomas Viehmann, 2020 (Manning Publications) - 一本提供PyTorch基础知识详细介绍的书,包括关于在CPU和GPU设备上管理张量和模型的章节和部分。