iuvppow 2022-01-22 22:12 采纳率: 100%
浏览 85
已结题

深度学习、Python、代码在服务器运行时遇到问题

本人的研究方向是图像去雾,在服务器运行代码时报出如下错误,搜索后发现找不到相关的解决方法,还请大家帮我看看,该怎么解决。

报错内容: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.

  • 写回答

2条回答 默认 最新

  • YirongChen 2022-01-23 07:51
    关注
    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))
    

    根据报错提示,上面这行代码疑似进行了矩阵的LU分解,而且该LU分解是对CUDA上的张量进行操作的。我估计是因为使用了torch.det函数,但传递给该函数的参数不满足这个条件“Calculates determinant of a square matrix or batches of square matrices”,参考:

    https://pytorch.org/docs/1.7.0/generated/torch.det.html

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论
查看更多回答(1条)

报告相同问题?

问题事件

  • 系统已结题 2月2日
  • 已采纳回答 1月25日
  • 创建了问题 1月22日
手机看
程序员都在用的中文IT技术交流社区

程序员都在用的中文IT技术交流社区

专业的中文 IT 技术社区,与千万技术人共成长

专业的中文 IT 技术社区,与千万技术人共成长

关注【CSDN】视频号,行业资讯、技术分享精彩不断,直播好礼送不停!

关注【CSDN】视频号,行业资讯、技术分享精彩不断,直播好礼送不停!

客服 返回
顶部