fix exp2 pytorch rewrite fatal train issue

tensorflow_gpu_util.py

@@ -0,0 +1,9 @@
+import tensorflow as tf
+
+def print_gpu_availability():
+    """打印Tensorflow的GPU可用性"""
+    gpus = tf.config.list_physical_devices('GPU')
+    if gpus:
+        print("GPU is available")
+    else:
+        print("GPU is not available")