我是 PyTorch 的菜鸟，我正在尝试训练我的模型（估计两个图像之间的仿射变换）。 该代码在 google colab 上运行。当我尝试在 cpu（没有设备 cuda）上运行它时，它可以工作，但是当...