tensorflow2.2程序，cpu模式下训练可以，换成gpu模式总是跳出？

一个深度学习模型，用tensorflow2.2的cpu版本，keras搭建的，训练没有任何问题。但是换成tensorflow-gpu2.2，除设置gpu模式外，其他没有变动，正常训练几分钟后总是跳出，很头疼，忘赐教。输出错误信息如下：
Tensorflow.python.framework.errors_impl.InvalidArgumentError: 2 root errors(s) found
(0)Invalid argument: assertion failed: [predictions must be >= 0] [Condition x >= y did not hold element-wise:] [x (model/conv2d_38/Sigmoid:0) = ] [[[[0.492345566 0.5051234456 0.4999924355]]]...] [y (Cast_6/x:0) = ] [0]
[[{{node assert_}.....................................
请教大家！谢谢

你安装的TensorFlow-gpu有问题，如果你是用pip安装的话，需要先安装cuda和cudnn，然后在用pip安装。如果你使用的是conda安装的话，原本是可以使用的，但是最近好像有问题，具体原因我也不是特别清楚。
推荐安装cuda和cudnn，然后安装TensorFlow-gpu。
如果有帮助，请采纳，多谢！