tensorflow cpu正常训练 gpu的loss nan

Question

使用anaconda虚拟环境建立的tf2.1gpu版本进行预测，loss函数采用mse。调用gpu进行运算显示loss=nan，而采用cpu计算loss为正常。

Answer 1

你好，我是有问必答小助手，非常抱歉，本次您提出的有问必答问题，技术专家团超时未为您做出解答


本次提问扣除的有问必答次数，将会以问答VIP体验卡（1次有问必答机会、商城购买实体图书享受95折优惠）的形式为您补发到账户。


因为有问必答VIP体验卡有效期仅有1天，您在需要使用的时候【私信】联系我，我会为您补发。