tensorflow 警告:根:更新GPU卡后在输入Tensor中发现NaN或Inf

thtygnil  于 2023-03-09  发布在  其他
关注(0)|答案(1)|浏览(600)

我试着在CTR-GCN上重新生成结果。给定的代码在最初运行良好。然后我把我的GPU卡换成RTX 3090ti。之后,安装所有的Nvidia驱动程序。但是现在,当我运行模型时,我得到“警告:根:在输入Tensor中发现NaN或Inf”这个警告,并使准确性非常低。
如果任何人遇到这种情况,请帮助我解决这个问题。

c9x0cxw0

c9x0cxw01#

当我从3070换成2060 Super时也遇到了完全相同的问题。当我换回3070时问题解决了。有趣的是,我有另一个模型在1080TI中训练,这也不适用于2060 Super,但适用于3070。

相关问题