python - 如何捕获内部 TensorFlow 错误：python 中的“GPU 同步失败”？

Question

TensorFlow 版本：1.x 和 2.x。

是一种在 python 中使用 try-expect 捕获错误的方法吗？

tensorflow.python.framework.errors_impl.InternalError: GPU sync failed

如何在不比较字符串的情况下检查“ e ”是否是“ GPU 同步失败”错误？

try:
   #do_something

except InternalError as e:
   if is_e_the_gpu_sync_failed(e):
      #do something

标签： pythontensorflowgpu