警告:根:更新GPU卡后，在输入张量中发现NaN或Inf

Question

我试图在CTR-GCN上重新生成结果。给定的代码最初运行良好。然后我把我的GPU卡换成了RTX 3090ti。之后，安装所有英伟达驱动程序。但现在，当我运行模型时；警告：根：在输入张量"中找到NaN或Inf；这种警告会使准确性非常低。

如果有人遇到这种情况，请一定帮我解决这个问题。

Answer 1

当我从3070变成2060 Super时，我也遇到了完全相同的问题。最终，直到我换回3070，这个问题才得以解决。有趣的是，我有另一个用1080ti训练的型号，这也不适用于2060 Super，但适用于3070。