ImportError: cannot import name ‘get_device‘ from ‘basicsr.utils.misc‘

把以下代码加入到文件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()

Guess you like

Origin blog.csdn.net/jiulinghouxiao/article/details/131128319