ESRGAN 问题 — "Torch not compiled with CUDA enabled"

ESRGAN Issue — "Torch not compiled with CUDA enabled"

需要帮助!尝试使用预训练模型 (ESRGAN) 运行 python 脚本,但每次我都会遇到此错误消息。可能是什么问题?

在此之前,我安装了以下依赖项:

通过稍微调整代码解决了问题,将设备类型从 'cuda' 更改为 'cpu'。

device = torch.device('cpu')