解决方法:
将使用torch.linalg模块或torch.inverse语句中变量移到CPU,或者用更旧的pytorch版本例如 PyTorch (1.10.0+cu111).
即torch.inverse(data)改为
device = data.device()
data=data.cpu()
data=torch.inverse(data)
data = data.to(device)
解决方法:
将使用torch.linalg模块或torch.inverse语句中变量移到CPU,或者用更旧的pytorch版本例如 PyTorch (1.10.0+cu111).
即torch.inverse(data)改为
device = data.device()
data=data.cpu()
data=torch.inverse(data)
data = data.to(device)