本地部署图片生成正常,但是图像处理(超分辨率)报错:ValueError: Expected a cuda device, but got: cpu
iwasnotyet opened this issue · comments
iwasnotyet commented
carefree0910 commented
确实是个问题,原因有两个:
empty_cuda_cache
时应该先看看当前device
是不是cpu
TranslatorAPI
也需要实现一个to
方法,从而在需要时可以把它放到cuda
上执行
我会尽快进行修复,修好后再让你试试😉
carefree0910 commented
刚刚把这个问题修复了,不过需要重新 pull 一下代码以及重新 pip install -e .
一下(因为需要更新背后 carefree-learn
的版本);如果还有问题的话,欢迎继续留言😆
iwasnotyet commented
刚刚把这个问题修复了,不过需要重新 pull 一下代码以及重新
pip install -e .
一下(因为需要更新背后carefree-learn
的版本);如果还有问题的话,欢迎继续留言😆
十分感谢!