tf2 的cuda memory问题

Author: Eden980429Created Apr 22, 2023Updated Apr 22, 2023

您好,tensorflow2官方github中的memory问题您有好的解决办法吗?具体可以看这个贴子。里面似乎说没有好的解决办法。我在跑代码时也总是遇到cuda initialization error,代码没跑完,运算就报错了。您有研究过这个问题吗?谢谢,可能说的不是很详细,希望您能给一点建议。

https://github.com/tensorflow/tensorflow/issues/36465

Source: lyhue1991/eat_tensorflow2_in_30_days