gradient - 没有 CUDA 的 Pytorch .backward() 方法

Question

我正在尝试在autograd 模块的Pytorch 教程中运行代码。但是，当我运行.backwards()呼叫时，出现错误：

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

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

Answer 1

您应该使用以下方法将网络、输入和标签传输到 cpu：net.cpu(), Variable(inputs.cpu()), Variable(labels.cpu())