没有CUDA的Pytorch .backward（）方法

Question

我正在尝试在autograd模块的Pytorch tutorial中运行代码。但是，当我运行.backwards()调用时，我收到错误：

cuda runtime error (38) : no CUDA-capable device is detected at torch/csrc/autograd/engine.cpp:359

我当然没有设置支持CUDA的设备，但我的理解是这并不是绝对必要的（至少我没有在教程中的任何地方找到它）。所以我想知道是否有一种方法可以在没有支持CUDA的GPU的情况下运行代码。

Answer 1

您应该使用：net.cpu(), Variable(inputs.cpu()), Variable(labels.cpu())

将网络，输入和标签传输到cpu上