ESRGAN 问题 — "Torch not compiled with CUDA enabled"
ESRGAN Issue — "Torch not compiled with CUDA enabled"
需要帮助!尝试使用预训练模型 (ESRGAN) 运行 python 脚本,但每次我都会遇到此错误消息。可能是什么问题?
在此之前,我安装了以下依赖项:
- pip 安装 pytorch cuda
- pip 安装 opencv-pythonglob2
通过稍微调整代码解决了问题,将设备类型从 'cuda' 更改为 'cpu'。
device = torch.device('cpu')
需要帮助!尝试使用预训练模型 (ESRGAN) 运行 python 脚本,但每次我都会遇到此错误消息。可能是什么问题?
在此之前,我安装了以下依赖项:
- pip 安装 pytorch cuda
- pip 安装 opencv-pythonglob2
通过稍微调整代码解决了问题,将设备类型从 'cuda' 更改为 'cpu'。
device = torch.device('cpu')