CUDA error: invalid configuration argument

[dongfengxijian]

你好，感谢分享你的工作。由于实验室只有一台3090（不匹配cuda10）的服务器，所以我只能安装pytorch1.7.0，python3.6.6，cudatoolkit11.0.3。但是会在main_branch.py的30行mat_inv_x, _ = torch.solve(eye_x, eye_x - (w ** 2) * affinity_mat)遇到CUDA error: invalid configuration argument这一报错。不知道该如何解决？求指导一下。

[LeonHLJ]

你好，感谢分享你的工作。由于实验室只有一台3090（不匹配cuda10）的服务器，所以我只能安装pytorch1.7.0，python3.6.6，cudatoolkit11.0.3。但是会在main_branch.py的30行mat_inv_x, _ = torch.solve(eye_x, eye_x - (w ** 2) * affinity_mat)遇到CUDA error: invalid configuration argument这一报错。不知道该如何解决？求指导一下。

For high versions of CUDA and Pytorch, you can try to change the Line30 of main_branch.py, i.e., "mat_inv_x, _ = torch.solve(eye_x, eye_x - (w ** 2) * affinity_mat" to "mat_inv_x = torch.inverse(eye_x - (w ** 2) * affinity_mat)". I have run our method in this way and obtain similar results.

[dongfengxijian]

@LeonHLJ 感谢大佬的指导，用这个方式可以在cuda11下运行。