本人的研究方向是图像去雾,在服务器运行代码时报出如下错误,搜索后发现找不到相关的解决方法,还请大家帮我看看,该怎么解决。
报错内容:File "train.py", line 226, in
gp_loss = gp_struct.compute_gploss(zy_in, imgid,batch_id,0)
File "/home/mist/GP.py", line 304, in compute_gploss
loss_unsup = torch.mean(((tensor_vec-mean_pred)*2)/sigma_est[0]) + 1.0self.lambda_vartorch.log(torch.det(sigma_est)) - 0.000001self.lambda_var*torch.log(torch.det(far_sigma_est))
RuntimeError: Calling torch.lu on a CUDA tensor requires compiling PyTorch with MAGMA. lease rebuild with MAGMA.