[环境配置]stable diffusion启动报错cannot import name ‘get_device‘ from ‘basicsr.utils.misc‘

python webui.py启动后报错cannot import name 'get_device' from 'basicsr.utils.misc'

找到Lib\site-packages\basicsr\utils\misc.py

在文件里面加入

import torch
def get_device():
    if torch.cuda.is_available():
        return torch.device("cuda")
    else:
        return torch.device("cpu")

def gpu_is_available():
        return torch.cuda.is_available()

猜你喜欢

转载自blog.csdn.net/FL1623863129/article/details/131133350