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